Home
last modified time | relevance | path

Searched defs:sub_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc59 const Shape& sub_shape = ShapeUtil::GetSubshape(literal_shape, leaf.first); in TransferLiteralToInfeed() local
Dir_emitter_unnested.cc392 const ShapeIndex& /*index*/) { in GetIndexTypeForKernel()
440 const ShapeIndex& /*index*/) { in GetIndexTypeForKernelFromMlir()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_liveness_analysis.cc99 [&](const Shape& sub_shape, const ShapeIndex& shape_index) { in MarkLiveAtAllIndices()
Dhlo_sharding.cc629 const Shape* sub_shape = &shape; in GetSubSharding() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc136 xla::Shape* sub_shape = in GetInfeedShapeWithLayout() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc219 const xla::ShapeIndex& index) -> Status { in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc963 ForEachSubshape(shape, [&](const Shape& sub_shape, const ShapeIndex& index) { in GetLeafShapes()
1592 const ShapeIndex& index) { in DynamicShapeIsCompatible()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1427 const Shape& sub_shape = in GetOrCreateView() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1970 SmallVector<int64_t, 2> sub_shape; in fold() local