| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 26 Status HloInputOutputAliasConfig::SetUpAlias(const ShapeIndex& output_index, in SetUpAlias()
88 ShapeIndex output_index(entry.output_shape_index().begin(), in CreateFromProto() local
109 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) { in ToString()
137 [&](const xla::ShapeIndex& output_index, absl::optional<Alias> alias) { in GetAliasedOutput()
155 [&](const ShapeIndex& output_index, absl::optional<Alias> aliased) { in ForEachAlias()
165 [&](const ShapeIndex& output_index, absl::optional<Alias> aliased) { in ForEachAliasWithStatus()
183 const Alias& alias) -> Status { in Verify()
|
| D | hlo_input_output_alias_config_test.cc | 38 void expect_aliased(const ShapeIndex& output_index, int64 param_number, in expect_aliased()
55 void expect_not_aliased(const ShapeIndex& output_index, int64 param_number, in expect_not_aliased()
|
| D | optimize_input_output_buffer_alias.cc | 73 const ShapeIndex& output_index = index; in Build() local
|
| /external/libtextclassifier/utils/ |
| D | tflite-model-executor.h | 114 TensorView<T> OutputView(const int output_index, in OutputView()
125 std::vector<T> Output(const int output_index, in Output()
|
| D | tflite-model-executor.cc | 226 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
239 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 82 int32 output_index; in InsertLogging() local
131 for (int output_index : node_outputs[node.name()]) { in InsertLogging() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 135 Status WriteOptionalWithValueToOutput(OpKernelContext* ctx, int output_index, in WriteOptionalWithValueToOutput()
147 Status WriteOptionalNoneToOutput(OpKernelContext* ctx, int output_index) { in WriteOptionalNoneToOutput()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.cc | 80 const Index output_index = in SortedSegmentSumCustomKernel() local
98 const Index output_index = in SortedSegmentSumCustomKernel() local
122 const Index output_index = in UnsortedSegmentCustomKernel() local
|
| D | ops_testutil.cc | 44 Tensor* OpsTestBase::GetOutput(int output_index) { in GetOutput()
|
| 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
|
| D | dynamic_partition_op.cc | 100 gtl::InlinedVector<int, 32> output_index(num_partitions_); in Compute() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | wrapped_tensor.h | 27 const int output_index; member
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execution_callbacks.py | 61 output_index, argument
113 def output_index(self): member in InfOrNanError
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.proto | 18 int64 output_index = 2; field
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | operation_jni.cc | 89 jint output_index) { in Java_org_tensorflow_Operation_shape()
142 jint output_index) { in Java_org_tensorflow_Operation_dtype()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 295 int output_index) { in BroadcastBinaryOpOutputShapeFn()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad.cc | 89 int output_index) { in Eval()
188 int output_index = 0; in Eval() local
|
| D | gather_nd.cc | 86 int output_index = 0; in Prepare() local
|
| D | gather.cc | 83 int output_index = 0; in Prepare() local
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 100 void InputIndexer::MoveToOutputIndex(tensorflow::int64 output_index) { in MoveToOutputIndex()
323 for (tensorflow::int64 output_index = start; output_index < limit; in do_periodic_resample_op() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | tensorrt_test.cc | 107 const int output_index = engine.getBindingIndex(kOutputTensor); in Execute() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 492 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
609 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
887 def _tensorflow_output_name(tensor_name, output_index): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 291 int output_index = in MapFunc() local
312 int output_index = in MapFunc() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | quantize.cc | 341 std::size_t output_index, ArrayDataType* quantized_data_type, in ChooseQuantizationForOperatorOutput()
622 for (std::size_t output_index = 0; output_index < op.outputs.size(); in Run() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.cc | 96 for (int output_index : info_->outputs()) { in Partition() local
|