Home
last modified time | relevance | path

Searched defs:shape_index (Results 1 – 22 of 22) 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.cc176 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary()
275 [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
Dhlo.proto356 repeated int64 shape_index = 3; field
Dlayout_assignment.cc1313 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
1350 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
Dhlo_instruction.cc3280 const ShapeIndex& shape_index) { in RelayoutConstant()
Dhlo_instructions.cc1045 const ShapeIndex& shape_index) { in RelayoutConstant()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h525 const Piece& piece(const ShapeIndex& shape_index) const { in piece()
697 Piece& piece(const ShapeIndex& shape_index) { in piece()
911 const ShapeIndex& shape_index) const { in data()
916 absl::Span<NativeT> MutableLiteralBase::data(const ShapeIndex& shape_index) { in data()
922 const ShapeIndex& shape_index) const { in Get()
933 const ShapeIndex& shape_index, in Set()
951 const ShapeIndex& shape_index) const { in GetSparseElement()
960 const ShapeIndex& shape_index) { in AppendSparseElement()
Dshape_layout.cc71 ShapeIndexView shape_index) { in ResetLayout()
Dliteral.cc204 const ShapeIndex& shape_index) { in sparse_indices()
973 void MutableLiteralBase::SortSparseElements(const ShapeIndex& shape_index) { in SortSparseElements()
1052 const ShapeIndex& shape_index, bool print_shape, in TupleToStringHelper()
1096 const ShapeIndex& shape_index, bool print_shape, in DenseArrayToStringHelper()
1168 void ToStringHelper(const LiteralBase& literal, const ShapeIndex& shape_index, in ToStringHelper()
2107 void* MutableLiteralBase::untyped_data(const ShapeIndex& shape_index) { in untyped_data()
Dshape_tree.h416 ShapeIndex shape_index = node->data.first; in InitChildren() local
452 ShapeIndex shape_index = node->data.first; in InitChildren() local
Dliteral_util.cc64 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()
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 <<()
Dliteral_comparison.cc741 const ShapeIndex& shape_index) { in NearHelper()
/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.cc154 ShapeIndexView shape_index, in GetTypedIrValue()
178 ShapeIndexView shape_index) { in BindHloToIrValue()
215 const ShapeIndex& shape_index) { in GetIrArray()
Dir_emitter_unnested.cc1214 ShapeIndex shape_index = in HandleSort() local
1329 ShapeIndex shape_index = in HandleSort() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h282 ShapeIndexView shape_index) { in ForceResultLayout()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc259 xla::ShapeIndex shape_index; in DoWork() local
Dxrt_state_ops.h337 xla::ShapeIndex shape_index; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc267 auto shape_index = tensorflow::ops::Placeholder(root, tensorflow::DT_INT32); in DestructureXrtAllocationTuple() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1340 int shape_index = IsSwitch(node) ? 0 : i; in UpdateOutputShapesUsingAnnotatedInformation() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc531 ShapeIndex shape_index = in HandleSort() local