| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 29 const ShapeIndex& output_index, int64_t param_number, in SetUpAlias()
83 ShapeIndex output_index(entry.output_shape_index().begin(), in CreateFromProto() local
103 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) { in ToString()
141 [&](const xla::ShapeIndex& output_index, absl::optional<Alias> alias) { in GetAliasedOutput()
160 [&](const ShapeIndex& output_index, absl::optional<Alias> aliased) { in ForEachAlias()
170 [&](const ShapeIndex& output_index, absl::optional<Alias> aliased) { in ForEachAliasWithStatus()
188 const Alias& alias) -> Status { in Verify()
|
| D | hlo_input_output_alias_config_test.cc | 37 void expect_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_aliased()
54 void expect_not_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_not_aliased()
|
| D | optimize_input_output_buffer_alias.cc | 75 const ShapeIndex& output_index = index; in Build() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.h | 85 const Index output_index = in SortedSegmentReductionCustomKernel() local
105 const Index output_index = in SortedSegmentReductionCustomKernel() local
128 const int64 output_index = in UnsortedSegmentCustomKernel() local
|
| D | reduce_join_op.cc | 50 int64_t output_index, const gtl::InlinedVector<int32, 8>& dim_list, in LinearSubIndexToFullIndex()
164 for (int64_t output_index = 0; output_index < output_shape.num_elements(); in Compute() local
|
| D | reshape_util_gpu.cu.cc | 38 Tindex* output_index = &output_indices[sparse_index * output_rank]; in ReshapeSparseTensorKernel() local
|
| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.h | 125 TensorView<T> OutputView(const int output_index, in OutputView()
136 std::vector<T> Output(const int output_index, in Output()
|
| D | tflite-model-executor.cc | 381 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
394 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 81 int32_t output_index; in InsertLogging() local
130 for (int output_index : node_outputs[node.name()]) { in InsertLogging() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_elemental_utils.cc | 119 ValueRange output_index, in elementalLower()
154 ValueRange output_index, in elementalLowerImplForBroadcastInDimOps()
207 ValueRange output_index, bool check_cache) { in elementalLower()
215 ValueRange output_index, in elementalLower()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 138 Status WriteOptionalWithValueToOutput(OpKernelContext* ctx, int output_index, in WriteOptionalWithValueToOutput()
150 Status WriteOptionalNoneToOutput(OpKernelContext* ctx, int output_index) { in WriteOptionalNoneToOutput()
|
| /external/XNNPACK/test/ |
| D | depth-to-space-operator-tester.h | 178 const size_t output_index = in TestNHWCxX32() local
244 const size_t output_index = in TestNCHW2NHWCxX32() local
|
| D | constant-pad-operator-tester.h | 156 const size_t output_index = in TestX8() local
271 const size_t output_index = in TestX16() local
386 const size_t output_index = in TestX32() local
|
| D | depthtospace-microkernel-tester.h | 131 const size_t output_index = in Test() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_operation_jni.cc | 83 jint output_index) { in Java_org_tensorflow_GraphOperation_shape()
133 jint output_index) { in Java_org_tensorflow_GraphOperation_dtype()
|
| /external/python/cffi/c/ |
| D | parse_c_type.c | 54 size_t output_index; member
813 int parse_c_type_from(struct _cffi_parse_info_s *info, size_t *output_index, in parse_c_type_from()
839 size_t output_index = 0; in parse_c_type() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 232 int output_index) { in BroadcastBinaryOpOutputShapeFn()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.proto | 19 int64 output_index = 2; field
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_depth.h | 67 const int output_index = in SpaceToDepth() local
|
| D | depth_to_space.h | 66 const int output_index = in DepthToSpace() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 178 int64 output_index = 2; field
188 int64 output_index = 1; field
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 91 auto output_index = xla::GetTupleElement(param, 4); in BuildInnerLoopBody() local
133 auto output_index = xla::GetTupleElement(inner_loop, 4); in BuildOuterLoopBody() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | buffer_reuse_pass.cc | 75 int32_t output_index = kIndexAmbiguous; in find_output_indices() local
250 if (auto output_index = analysis.get_output_index(op)) { in runOnFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 376 for (int output_index = 0; output_index < cluster_func_output_index; in MapClusterOutputIndexWithRegionOutputIndex() local
507 const auto output_index = result_and_index.index(); in GetOutputTypesForLogicalDeviceComputation() local
543 const auto output_index = result_and_index.index(); in RemapOutputsFromLogicalDevices() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 499 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
619 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
903 def _tensorflow_output_name(tensor_name, output_index): argument
|