Home
last modified time | relevance | path

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

/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.cc197 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary()
291 shape, [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
Dlayout_assignment.cc768 [&](const Shape& subshape, const ShapeIndex& shape_index) { in CheckParameterLayout()
1326 const auto shape_index = in PropagateOperandConstraint() local
1395 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
1432 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
2016 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateComputationLayouts()
Dhlo_cost_analysis.cc776 [this, fusion](const Shape& subshape, const ShapeIndex& shape_index) { in HandleFusion()
815 const ShapeIndex& shape_index) { in HandleFusion()
Dhlo.proto388 repeated int64 shape_index = 3; field
Dmemory_space_assignment.cc1246 ShapeIndex shape_index) { in ReplaceTupleWith()
Dhlo_instruction.cc3613 const ShapeIndex& shape_index) { in RelayoutConstant()
Dhlo_instructions.cc1144 const ShapeIndex& shape_index) { in RelayoutConstant()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Ddynamic_padding.proto14 int32 shape_index = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc163 ShapeIndexView shape_index, in GetTypedIrValue()
188 ShapeIndexView shape_index) { in BindHloToIrValue()
225 const ShapeIndex& shape_index) { in GetIrArray()
Dir_emitter_unnested.cc1035 ShapeIndex shape_index = in HandleSort() local
1150 ShapeIndex shape_index = in HandleSort() local
3203 ShapeIndex shape_index = (slice_or_tuple->opcode() == HloOpcode::kSlice) in EmitElementForInputFusibleSlices() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h507 const Piece& piece(const ShapeIndex& shape_index) const { in piece()
660 Piece& piece(const ShapeIndex& shape_index) { in piece()
873 const ShapeIndex& shape_index) const { in data()
878 absl::Span<NativeT> MutableLiteralBase::data(const ShapeIndex& shape_index) { in data()
884 const ShapeIndex& shape_index) const { in Get()
895 const ShapeIndex& shape_index, in Set()
Dshape_layout.cc76 ShapeIndexView shape_index) { in ResetLayout()
Dliteral_comparison.cc294 const ShapeIndex& shape_index, ErrorSpec error, in Compare()
330 const ShapeIndex& shape_index, ErrorSpec error, in NearComparator()
689 const ShapeIndex& shape_index, in EqualHelper()
785 const ShapeIndex& shape_index, const ErrorSpec& error, in NearHelper()
Dshape_tree.h419 ShapeIndex shape_index = node->data.first; in InitChildren() local
455 ShapeIndex shape_index = node->data.first; in InitChildren() local
Dliteral_util.cc64 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()
Dliteral.cc973 const ShapeIndex& shape_index, bool print_shape, in TupleToStringHelper()
991 const ShapeIndex& shape_index, bool print_shape, in DenseArrayToStringHelper()
1063 void ToStringHelper(const LiteralBase& literal, const ShapeIndex& shape_index, in ToStringHelper()
1971 void* MutableLiteralBase::untyped_data(const ShapeIndex& shape_index) { in untyped_data()
Dshape_util.cc68 std::ostream& operator<<(std::ostream& out, const ShapeIndex& shape_index) { in operator <<()
73 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h260 ShapeIndexView shape_index) { in ForceResultLayout()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h385 xla::ShapeIndex shape_index; in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1085 int arg_index, int32_t shape_index, in AddDynamicParameterBindingEntry()
1154 const int32_t shape_index = shape_index_attr.getInt(); in AddDynamicParameterBindings() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1433 int shape_index = IsSwitch(node) ? 0 : i; in UpdateOutputShapesUsingAnnotatedInformation() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc590 ShapeIndex shape_index = in HandleSort() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc90 int shape_index, absl::string_view attr_value) { in VerifyShapeAttributeMatch()