| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 29 const ShapeIndex& output_index, int64 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 param_number, in expect_aliased()
54 void expect_not_aliased(const ShapeIndex& output_index, int64 param_number, in expect_not_aliased()
|
| D | optimize_input_output_buffer_alias.cc | 75 const ShapeIndex& output_index = index; in Build() 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 | 359 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
372 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 81 int32 output_index; in InsertLogging() local
130 for (int output_index : node_outputs[node.name()]) { in InsertLogging() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.cc | 80 const Index output_index = in SortedSegmentReductionCustomKernel() local
100 const Index output_index = in SortedSegmentReductionCustomKernel() local
123 const int64 output_index = in UnsortedSegmentCustomKernel() local
|
| D | reduce_join_op.cc | 50 int64 output_index, const gtl::InlinedVector<int32, 8>& dim_list, in LinearSubIndexToFullIndex()
164 for (int64 output_index = 0; output_index < output_shape.num_elements(); in Compute() local
|
| /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/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | wrapped_tensor.h | 27 const int output_index; member
|
| /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 | 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 | 229 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 | 65 const int output_index = in SpaceToDepth() local
|
| D | depth_to_space.h | 66 const int output_index = in DepthToSpace() local
|
| D | prelu.h | 45 int output_index = Offset(extended_output_shape, b, y, x, c); in BroadcastPrelu4DSlow() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 41 int32_t output_index, in _mlir_ciface_tf_alloc()
|
| /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/tensorflow/utils/ |
| D | xla_sharding_util.cc | 378 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 | 506 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
626 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
911 def _tensorflow_output_name(tensor_name, output_index): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 94 int output_index = 0; in Prepare() local
|