Searched refs:input_element_shape (Results 1 – 2 of 2) sorted by relevance
956 TensorShape input_element_shape = input.shape(); in Compute() local957 input_element_shape.RemoveDim(0); in Compute()966 c, l->element_shape.IsCompatibleWith(input_element_shape), in Compute()970 b, ". Op element shape: ", input_element_shape.DebugString(), in Compute()1009 element_dtype_, input_element_shape, &tmp, &frame)); in Compute()1010 if (input_element_shape.num_elements() > 0) { in Compute()
129 const Shape input_element_shape = in VerifyReducerShape() local132 input_element_shape, reducer_shape.parameters(inputs + i))) { in VerifyReducerShape()138 ShapeUtil::HumanString(input_element_shape)); in VerifyReducerShape()