Home
last modified time | relevance | path

Searched refs:input_element_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h945 TensorShape input_element_shape = input.shape(); in Compute() local
946 input_element_shape.RemoveDim(0); in Compute()
955 c, l->element_shape.IsCompatibleWith(input_element_shape), in Compute()
959 b, ". Op element shape: ", input_element_shape.DebugString(), in Compute()
998 element_dtype_, input_element_shape, &tmp, &frame)); in Compute()
999 if (input_element_shape.num_elements() > 0) { in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc131 const Shape input_element_shape = in VerifyReducerShape() local
134 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()
140 ShapeUtil::HumanString(input_element_shape)); in VerifyReducerShape()