Home
last modified time | relevance | path

Searched refs:dim_size_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h258 const Shape& operand, absl::Span<const Shape* const> dim_size_shapes,
Dshape_inference.cc2912 const Shape& operand, absl::Span<const Shape* const> dim_size_shapes, in InferDynamicReshapeShape() argument
2922 for (const Shape* dim_size_shape : dim_size_shapes) { in InferDynamicReshapeShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1154 TF_ASSIGN_OR_RETURN(const auto& dim_size_shapes, in DynamicReshape()
1157 absl::c_transform(dim_size_shapes, std::back_inserter(dim_size_shape_ptrs), in DynamicReshape()