private void RunGraphsAndCompareOutputs(TF_Output[] grad_outputs, TF_Output[] expected_grad_outputs) { var csession = new CSession(graph_, s_); var expected_csession = new CSession(expected_graph_, s_); var grad_outputs_vec = grad_outputs; csession.SetOutputs(grad_outputs_vec); csession.Run(s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)); var out0 = csession.output_tensor(0); var out1 = csession.output_tensor(1); var expected_grad_outputs_vec = expected_grad_outputs; expected_csession.SetOutputs(expected_grad_outputs_vec); expected_csession.Run(s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)); var expected_out0 = expected_csession.output_tensor(0); var expected_out1 = expected_csession.output_tensor(1); //CompareTensors(out0, expected_out0); //CompareTensors(out1, expected_out1); }
unsafe void Run(KeyValuePair <Operation, Tensor>[] inputs, TF_Output[] outputs, int[] expected_results) { var csession = new CSession(host_graph_, s_); ASSERT_EQ(TF_OK, s_.Code, s_.Message); csession.SetInputs(inputs); csession.SetOutputs(outputs); csession.Run(s_); ASSERT_EQ(TF_OK, s_.Code, s_.Message); for (int i = 0; i < expected_results.Length; ++i) { var output = csession.output_tensor(i); ASSERT_TRUE(!output.IsInvalid); EXPECT_EQ(TF_DataType.TF_INT32, c_api.TF_TensorType(output)); EXPECT_EQ(0, c_api.TF_NumDims(output)); ASSERT_EQ(sizeof(int), (int)c_api.TF_TensorByteSize(output)); var output_contents = c_api.TF_TensorData(output); EXPECT_EQ(expected_results[i], *(int *)output_contents.ToPointer()); } }