Searched refs:is_tensor_list (Results 1 – 11 of 11) sorted by relevance
50 self.assertFalse(tensors.is_tensor_list(self._simple_tensor_array()))51 self.assertTrue(tensors.is_tensor_list(self._simple_tensor_list()))52 self.assertFalse(tensors.is_tensor_list(constant_op.constant(1)))53 self.assertFalse(tensors.is_tensor_list(self._simple_list_of_tensors()))54 self.assertFalse(tensors.is_tensor_list(None))
42 def is_tensor_list(t): function
127 then_result.outputs[output_index].is_tensor_list; in ValidateShapes()129 else_result.outputs[output_index].is_tensor_list; in ValidateShapes()308 if (then_result.outputs[i].is_tensor_list) { in Compile()
250 branch_results[0].outputs[output_index].is_tensor_list; in Compile()252 branch_results[j].outputs[output_index].is_tensor_list; in Compile()324 if (branch_results[0].outputs[i].is_tensor_list) { in Compile()
115 bool is_tensor_list = false; member
305 if (result.outputs[i].is_tensor_list) { in CompileFunctionalNode()
1634 ASSERT_TRUE(output0.is_tensor_list); in TEST_F()1636 ASSERT_TRUE(output1.is_tensor_list); in TEST_F()
214 output.is_tensor_list = true; in BuildComputation()
102 if (xla_output_description.is_tensor_list) output << " tensor_list"; in PrintHloModuleText()
194 out_desc.is_tensor_list = false; in GetOutputInfo()
246 elif tensors.is_tensor_list(s):