Searched refs:dim_shape (Results 1 – 8 of 8) sorted by relevance
109 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in CheckOutputSize() local110 dim_shape = dim_shape < 0 ? 0 : dim_shape; in CheckOutputSize()112 TF_LITE_ENSURE_EQ(context, output_shape->data[shape_size], dim_shape); in CheckOutputSize()
127 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in ResizeOutputTensor() local128 dim_shape = dim_shape < 0 ? 0 : dim_shape; in ResizeOutputTensor()130 output_shape_vector.push_back(dim_shape); in ResizeOutputTensor()
234 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile() local241 dim_shape.DebugString()))); in Compile()
237 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local239 dim_shape->set_size(Value(dim)); in ShapeHandleToProto()241 dim_shape->set_size(-1); in ShapeHandleToProto()
599 dim_shape = array_ops.stack(601 batch_indices += array_ops.reshape(dim_indices, dim_shape)
5172 dim_shape = stack(5174 batch_indices += reshape(dim_indices, dim_shape)
2212 const Shape& dim_shape = ShapeUtil::GetSubshape(hlo->shape(), {i}); in HandlePadToStatic() local2213 TF_RET_CHECK(Shape::Equal()(dim_shape, ShapeUtil::MakeScalarShape(S32))); in HandlePadToStatic()
855 const Shape& dim_shape = TypeToShape(pad_to_static.output()[i].getType()); in EmitPadToStaticFromMlir() local856 TF_RET_CHECK(Shape::Equal()(dim_shape, ShapeUtil::MakeScalarShape(S32))); in EmitPadToStaticFromMlir()