Home
last modified time | relevance | path

Searched defs:shape_index (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_liveness_analysis.cc54 index_tree.ForEachElement([&](const ShapeIndex& shape_index, bool live) { in ForEachLiveIndex()
66 const ShapeIndex& shape_index, in MarkLiveAtIndex()
122 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughTuple()
153 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughGTE()
172 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughWhile()
202 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessToParameterCallers()
255 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughControlFlow()
Dtuple_util.cc63 ShapeIndex shape_index, bool insert_bitcast_if_different_shape) { in ReplaceTupleWith()
Dlayout_assignment.cc931 [&](const Shape& subshape, const ShapeIndex& shape_index) { in CheckParameterLayout()
1558 const auto shape_index = in PropagateOperandConstraint() local
1645 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
1680 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
2372 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateComputationLayouts()
Dinstruction_fusion.cc215 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary()
311 shape, [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
Dhlo_cost_analysis.cc891 [this, fusion](const Shape& subshape, const ShapeIndex& shape_index) { in HandleFusion()
930 const ShapeIndex& shape_index) { in HandleFusion()
Dhlo.proto535 repeated int64 shape_index = 3; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc142 ShapeIndexView shape_index, in GetTypedIrValue()
157 ShapeIndexView shape_index) { in BindHloToIrValue()
172 const ShapeIndex& shape_index) { in GetIrArray()
Dinfeed_thunk.cc44 const ShapeIndex& shape_index = source.first; in ExecuteOnStream() local
Doutfeed_thunk.cc57 const ShapeIndex& shape_index = output_leaf_it->first; in ExecuteOnStream() local
Dgpu_executable.cc902 xla::ShapeIndex shape_index; in SetUpMlirAllocation() local
924 ShapeIndex shape_index; in SetUpMlirAllocation() local
/external/tensorflow/tensorflow/core/protobuf/tpu/
Ddynamic_padding.proto14 int32 shape_index = 2; field
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h747 const Piece& piece(const ShapeIndex& shape_index) const { in piece()
912 Piece& piece(const ShapeIndex& shape_index) { in piece()
1237 const ShapeIndex& shape_index) const { in data()
1242 absl::Span<NativeT> MutableLiteralBase::data(const ShapeIndex& shape_index) { in data()
1248 const ShapeIndex& shape_index) const { in Get()
1259 const ShapeIndex& shape_index, in Set()
Dshape_layout.cc100 ShapeIndexView shape_index) { in ResetLayout()
Dliteral_comparison.cc286 const ShapeIndex& shape_index, ErrorSpec error, in Compare()
322 const ShapeIndex& shape_index, ErrorSpec error, in NearComparator()
685 const ShapeIndex& shape_index, in EqualHelper()
781 const ShapeIndex& shape_index, const ErrorSpec& error, in NearHelper()
Dliteral.cc511 Literal Literal::SubLiteral(ShapeIndexView shape_index) { in SubLiteral()
674 const ShapeIndex& shape_index, in SetDynamicSize()
1302 const ShapeIndex& shape_index, bool print_shape, in TupleToStringHelper()
1322 const ShapeIndex& shape_index, bool print_shape, in DenseArrayToStringHelper()
1410 void ToStringHelper(const LiteralBase& literal, const ShapeIndex& shape_index, in ToStringHelper()
2539 void* MutableLiteralBase::untyped_data(const ShapeIndex& shape_index) { in untyped_data()
Dliteral_util.cc63 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()
Dshape_util.cc86 std::ostream& operator<<(std::ostream& out, const ShapeIndex& shape_index) { in operator <<()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h318 ShapeIndexView shape_index) { in ForceResultLayout()
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dmemory_viewer_preprocess.proto39 repeated int64 shape_index = 5; field
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_tfrt_cpu_device_buffer.cc114 const ShapeIndex& shape_index) { in Buffer()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc501 xla::ShapeIndex* shape_index, OpBuilder* b, Location loc) { in RewriteFusionOperand()
561 xla::ShapeIndex shape_index; in EmitFusionOp() local
1428 const xla::ShapeIndex& shape_index) { in GetOrCreateArrayView()
1536 xla::ShapeIndex shape_index = result_subset; in GetOrCreateView() local
1667 const xla::ShapeIndex& shape_index = iter->second.second; in Initialize() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_conversions.cc406 void Destroy(XLA_ShapeIndex* shape_index) { delete[] shape_index; } in Destroy()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h387 xla::ShapeIndex shape_index; in Compute() local
/external/tensorflow/tensorflow/core/profiler/convert/
Dhlo_proto_to_memory_visualization_utils.cc138 absl::Span<const int64_t> shape_index) { in ResolveShapeIndex()
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc200 ShapeIndex shape_index = {}; member
242 ShapeIndex shape_index; member

12