/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_liveness_analysis.cc | 55 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()
|
D | instruction_fusion.cc | 176 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary() 275 [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
|
D | hlo.proto | 356 repeated int64 shape_index = 3; field
|
D | layout_assignment.cc | 1313 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint() 1350 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
|
D | hlo_instruction.cc | 3280 const ShapeIndex& shape_index) { in RelayoutConstant()
|
D | hlo_instructions.cc | 1045 const ShapeIndex& shape_index) { in RelayoutConstant()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | literal.h | 525 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()
|
D | shape_layout.cc | 71 ShapeIndexView shape_index) { in ResetLayout()
|
D | literal.cc | 204 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()
|
D | shape_tree.h | 416 ShapeIndex shape_index = node->data.first; in InitChildren() local 452 ShapeIndex shape_index = node->data.first; in InitChildren() local
|
D | literal_util.cc | 64 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()
|
D | shape_util.cc | 68 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 <<()
|
D | literal_comparison.cc | 741 const ShapeIndex& shape_index) { in NearHelper()
|
/external/tensorflow/tensorflow/core/protobuf/tpu/ |
D | dynamic_padding.proto | 14 int32 shape_index = 2; field
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | hlo_to_ir_bindings.cc | 154 ShapeIndexView shape_index, in GetTypedIrValue() 178 ShapeIndexView shape_index) { in BindHloToIrValue() 215 const ShapeIndex& shape_index) { in GetIrArray()
|
D | ir_emitter_unnested.cc | 1214 ShapeIndex shape_index = in HandleSort() local 1329 ShapeIndex shape_index = in HandleSort() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | hlo_test_base.h | 282 ShapeIndexView shape_index) { in ForceResultLayout()
|
/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | xrt_execute_op.cc | 259 xla::ShapeIndex shape_index; in DoWork() local
|
D | xrt_state_ops.h | 337 xla::ShapeIndex shape_index; in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xrt.cc | 267 auto shape_index = tensorflow::ops::Placeholder(root, tensorflow::DT_INT32); in DestructureXrtAllocationTuple() local
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 1340 int shape_index = IsSwitch(node) ? 0 : i; in UpdateOutputShapesUsingAnnotatedInformation() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 531 ShapeIndex shape_index = in HandleSort() local
|