Searched refs:dim_shape (Results 1 – 4 of 4) sorted by relevance
123 int32_t dim_shape = ceil((end - begin) / static_cast<float>(stride)); in ResizeOutputTensor() local124 dim_shape = dim_shape < 0 ? 0 : dim_shape; in ResizeOutputTensor()126 output_shape_vector.push_back(dim_shape); in ResizeOutputTensor()
120 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile() local127 dim_shape.DebugString()))); in Compile()
294 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local296 dim_shape->set_size(Value(dim)); in ShapeHandleToProto()298 dim_shape->set_size(-1); in ShapeHandleToProto()
3506 dim_shape = stack(3508 batch_indices += reshape(dim_indices, dim_shape)