Home
last modified time | relevance | path

Searched defs:shape_index (Results 1 – 25 of 28) 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.cc205 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyAtMostUnary()
299 shape, [&size](const Shape& subshape, const ShapeIndex& shape_index) { in ComputeGloballyUnfusible()
Dlayout_assignment.cc769 [&](const Shape& subshape, const ShapeIndex& shape_index) { in CheckParameterLayout()
1327 const auto shape_index = in PropagateOperandConstraint() local
1410 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
1451 user->shape(), [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateOperandConstraint()
2056 [&](const Shape& subshape, const ShapeIndex& shape_index) { in PropagateComputationLayouts()
Dhlo_cost_analysis.cc836 [this, fusion](const Shape& subshape, const ShapeIndex& shape_index) { in HandleFusion()
875 const ShapeIndex& shape_index) { in HandleFusion()
Dhlo.proto414 repeated int64 shape_index = 3; field
Dmemory_space_assignment.cc2763 ShapeIndex shape_index) { in ReplaceTupleWith()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc141 ShapeIndexView shape_index, in GetTypedIrValue()
156 ShapeIndexView shape_index) { in BindHloToIrValue()
193 const ShapeIndex& shape_index) { in GetIrArray()
Dinfeed_thunk.cc47 const ShapeIndex& shape_index = source.first; in ExecuteOnStream() local
Doutfeed_thunk.cc60 const ShapeIndex& shape_index = output_leaf_it->first; in ExecuteOnStream() local
Dgpu_compiler.cc1057 ShapeIndex shape_index; in GetMlirAllocationInfo() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h559 const Piece& piece(const ShapeIndex& shape_index) const { in piece()
719 Piece& piece(const ShapeIndex& shape_index) { in piece()
936 const ShapeIndex& shape_index) const { in data()
941 absl::Span<NativeT> MutableLiteralBase::data(const ShapeIndex& shape_index) { in data()
947 const ShapeIndex& shape_index) const { in Get()
958 const ShapeIndex& shape_index, in Set()
Dshape_layout.cc76 ShapeIndexView shape_index) { in ResetLayout()
Dliteral_comparison.cc283 const ShapeIndex& shape_index, ErrorSpec error, in Compare()
319 const ShapeIndex& shape_index, ErrorSpec error, in NearComparator()
678 const ShapeIndex& shape_index, in EqualHelper()
774 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.cc533 const ShapeIndex& shape_index, in SetDynamicSize()
1169 const ShapeIndex& shape_index, bool print_shape, in TupleToStringHelper()
1187 const ShapeIndex& shape_index, bool print_shape, in DenseArrayToStringHelper()
1273 void ToStringHelper(const LiteralBase& literal, const ShapeIndex& shape_index, in ToStringHelper()
2253 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.cc97 std::ostream& operator<<(std::ostream& out, const ShapeIndex& shape_index) { in operator <<()
102 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/tests/
Dhlo_test_base.h283 ShapeIndexView shape_index) { in ForceResultLayout()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc448 xla::ShapeIndex* shape_index, OpBuilder* b, Location loc) { in RewriteFusionOperand()
489 xla::ShapeIndex shape_index; in EmitFusionOp() local
1331 const xla::ShapeIndex& shape_index) { in GetOrCreateArrayView()
1426 xla::ShapeIndex shape_index = result_subset; in GetOrCreateView() 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.h385 xla::ShapeIndex shape_index; in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1644 int arg_index, int32_t shape_index, in AddDynamicParameterBindingEntry()
1713 const int32_t shape_index = shape_index_attr.getInt(); in AddDynamicParameterBindings() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1611 int shape_index = IsSwitch(node) ? 0 : i; in UpdateOutputShapesUsingAnnotatedInformation() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc556 ShapeIndex shape_index = in HandleSort() local

12