Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_liveness_analysis.cc55 index_tree.ForEachElement([&](const ShapeIndex& shape_index, bool live) { in ForEachLiveIndex()
67 const ShapeIndex& shape_index, in MarkLiveAtIndex()
99 [&](const Shape& sub_shape, const ShapeIndex& shape_index) { in MarkLiveAtAllIndices()
126 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughTuple()
157 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughGTE()
176 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessThroughWhile()
207 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) { in PropagateLivenessToParameterCallers()
Dinstruction_fusion.cc210 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary()
304 shape, [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
Dlayout_assignment.cc856 [&](const Shape& subshape, const ShapeIndex& shape_index) { in CheckParameterLayout()
1415 const auto shape_index = in PropagateOperandConstraint() local
1498 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
1539 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
2183 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateComputationLayouts()
Dhlo_cost_analysis.cc865 [this, fusion](const Shape& subshape, const ShapeIndex& shape_index) { in HandleFusion()
904 const ShapeIndex& shape_index) { in HandleFusion()
Dhlo.proto462 repeated int64 shape_index = 3; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc141 ShapeIndexView shape_index, in GetTypedIrValue()
156 ShapeIndexView shape_index) { in BindHloToIrValue()
171 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_compiler.cc1222 xla::ShapeIndex shape_index; in GetMlirAllocationInfo() local
1243 ShapeIndex shape_index; in GetMlirAllocationInfo() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h566 const Piece& piece(const ShapeIndex& shape_index) const { in piece()
731 Piece& piece(const ShapeIndex& shape_index) { in piece()
952 const ShapeIndex& shape_index) const { in data()
957 absl::Span<NativeT> MutableLiteralBase::data(const ShapeIndex& shape_index) { in data()
963 const ShapeIndex& shape_index) const { in Get()
974 const ShapeIndex& shape_index, in Set()
Dshape_layout.cc100 ShapeIndexView shape_index) { in ResetLayout()
Dliteral_comparison.cc285 const ShapeIndex& shape_index, ErrorSpec error, in Compare()
321 const ShapeIndex& shape_index, ErrorSpec error, in NearComparator()
681 const ShapeIndex& shape_index, in EqualHelper()
777 const ShapeIndex& shape_index, const ErrorSpec& error, in NearHelper()
Dshape_tree.h458 ShapeIndex shape_index = node->data.first; in InitChildren() local
494 ShapeIndex shape_index = node->data.first; in InitChildren() local
Dliteral.cc409 Literal Literal::SubLiteral(ShapeIndexView shape_index) { in SubLiteral()
547 const ShapeIndex& shape_index, in SetDynamicSize()
1183 const ShapeIndex& shape_index, bool print_shape, in TupleToStringHelper()
1201 const ShapeIndex& shape_index, bool print_shape, in DenseArrayToStringHelper()
1287 void ToStringHelper(const LiteralBase& literal, const ShapeIndex& shape_index, in ToStringHelper()
2362 void* MutableLiteralBase::untyped_data(const ShapeIndex& shape_index) { in untyped_data()
Dliteral_util.cc64 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()
Dshape_util.cc99 std::ostream& operator<<(std::ostream& out, const ShapeIndex& shape_index) { in operator <<()
104 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Ddynamic_padding.proto14 int32 shape_index = 2; field
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_tfrt_cpu_device_buffer.cc52 const ShapeIndex& shape_index) { in Buffer()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h286 ShapeIndexView shape_index) { in ForceResultLayout()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc481 xla::ShapeIndex* shape_index, OpBuilder* b, Location loc) { in RewriteFusionOperand()
540 xla::ShapeIndex shape_index; in EmitFusionOp() local
1374 const xla::ShapeIndex& shape_index) { in GetOrCreateArrayView()
1487 xla::ShapeIndex shape_index = result_subset; in GetOrCreateView() local
1627 const xla::ShapeIndex& shape_index = iter->second.second; in Initialize() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc366 void Free(XLA_ShapeIndex* shape_index) { delete[] shape_index; } in Free()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h387 xla::ShapeIndex shape_index; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc194 ShapeIndex shape_index = {}; member
236 ShapeIndex shape_index; member
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1783 int arg_index, int32_t shape_index, in AddDynamicParameterBindingEntry()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1638 int shape_index = IsSwitch(node) ? 0 : i; in UpdateOutputShapesUsingAnnotatedInformation() local

12