Home
last modified time | relevance | path

Searched defs:dynamic_dimensions (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape.h43 absl::Span<const bool> dynamic_dimensions, in Shape()
87 absl::Span<const bool> dynamic_dimensions() const { in dynamic_dimensions() function
Dshape_util.cc265 const std::vector<bool>& dynamic_dimensions) { in MakeShape()
290 const std::vector<bool>& dynamic_dimensions) { in MakeValidatedShape()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc120 std::vector<bool> dynamic_dimensions(rank, false); in TensorShapeToXLAShape() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtransform_unranked_hlo.cc380 auto dynamic_dimensions = llvm::SmallVector<int64_t, 6>( in createRankSpecializedBroadcastAndOp() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h222 BoolList dynamic_dimensions; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc4417 std::vector<bool>* dynamic_dimensions) { in ParseDimensionSizes()
4595 std::vector<bool> dynamic_dimensions; in ParseShape() local