Searched refs:input_element_shape (Results 1 – 2 of 2) sorted by relevance
945 TensorShape input_element_shape = input.shape(); in Compute() local946 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()
131 const Shape input_element_shape = in VerifyReducerShape() local134 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()140 ShapeUtil::HumanString(input_element_shape)); in VerifyReducerShape()