| /external/tensorflow/tensorflow/core/common_runtime/ |
| memory_types.cc | 31 int output_index; member in struct:tensorflow::Endpoint
37 x.output_index);
43 return (x.node_id == y.node_id) && (x.output_index == y.output_index);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| deep_conv2d_test.cc | 31 const int output_index = k * cols * cols + l; local
32 matrix_out[output_index_base + output_index] =
|
| segment_reduction_ops_gpu.cu.cc | 77 const Index output_index = local
82 CudaAtomicAdd(output + output_index, sum);
84 *(output + output_index) = sum;
95 const Index output_index = local
97 CudaAtomicAdd(output + output_index, sum);
119 const Index output_index = local
121 KernelReductionFunctor()(output + output_index, ldg(input + input_index));
|
| sparse_cross_op.cc | 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; local
153 indices_matrix(output_index, 0) = batch_index;
154 indices_matrix(output_index, 1) = cross_count;
157 value_vec(output_index) = cross;
|
| conv_ops_gpu_3.cu.cc | 184 int output_index = index; local
186 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
195 output[output_index] =
211 int output_index = index; local
212 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
221 output[output_index] =
351 int output_index = output_origin_flat_index + ti * output_dims[2] + tj; local
357 output[output_index] = shared_memory_tile[tj][i_loc];
358 output_index += output_increment;
363 output[output_index] = shared_memory_tile[tj][i_loc]
379 int output_index = index; local
408 int output_index = index; local
[all...] |
| set_kernels.cc | 281 const auto output_index = std::inner_product( local
283 out(output_index) = group_set.size();
|
| deep_conv2d.cc | 821 const int64 output_index = local
[all...] |
| /external/tensorflow/tensorflow/python/eager/ |
| execution_callbacks.py | 42 output_index,
52 output_index: The 0-based output index of the tensor that contains
59 self._output_index = output_index
94 def output_index(self): member in class:InfOrNanError
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| gather.cc | 70 int output_index = 0; local
72 output_shape->data[output_index++] = input->dims->data[i];
75 output_shape->data[output_index++] = positions->dims->data[i];
78 output_shape->data[output_index++] = input->dims->data[i];
|
| /external/tensorflow/tensorflow/contrib/tensorrt/ |
| tensorrt_test.cc | 105 const int output_index = engine.getBindingIndex(kOutputTensor); local
110 ASSERT_EQ(0, cudaMalloc(&buffers[output_index], sizeof(float)));
122 ASSERT_EQ(0, cudaMemcpyAsync(output, buffers[output_index], sizeof(float),
129 ASSERT_EQ(0, cudaFree(buffers[output_index]));
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| parse_testdata.cc | 195 int output_index = interpreter->outputs()[i]; local
196 if (const float* data = interpreter->typed_tensor<float>(output_index)) {
217 interpreter->typed_tensor<int32_t>(output_index)) {
229 interpreter->typed_tensor<int64_t>(output_index)) {
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| import.cc | 136 auto output_index = outputs->Get(i); local
137 const string& output_name = tensors_table.at(output_index);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| scan_dataset_op.cc | 192 const size_t output_index = i - dataset()->state_types_.size(); variable
193 if (state_and_output[i].dtype() != output_dtypes()[output_index]) {
197 DataTypeString(dataset()->state_types_[output_index]),
200 if (!output_shapes()[output_index].IsCompatibleWith(
204 " (expected ", output_shapes()[output_index].DebugString(),
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| insert_logging.cc | 82 int32 output_index; local
83 if (!strings::safe_strto32(output_index_string, &output_index)) {
87 node_outputs[name].insert(output_index);
131 for (int output_index : node_outputs[node.name()]) {
132 print_node->add_input(strings::StrCat(node.name(), ":", output_index));
|
| /external/mesa3d/src/gallium/state_trackers/nine/ |
| nine_shader.h | 94 int output_index; member in struct:nine_vs_output_info
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla.cc | 70 const int output_index = 0; local
92 .Attr("T", BaseType(feed_node->output_type(output_index)))
106 if (edge->src_output() == output_index) {
130 if (id.output_index() >= fetch_node->num_outputs()) {
139 .Input(fetch_node, id.output_index())
140 .Attr("T", BaseType(fetch_node->output_type(id.output_index())))
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| subgraph.py | 570 def output_index(self, t): member in class:SubGraphView
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | 393 const int output_index = local
397 output_data[output_index] = input_data[input_index];
434 const int output_index = local
438 output_data[output_index] = input_data[input_index];
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| sparse_feature_cross_kernel.cc | 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; local
153 indices_matrix(output_index, 0) = batch_index;
154 indices_matrix(output_index, 1) = cross_count;
157 value_vec(output_index) = cross;
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils.cc | 174 const int output_index = input_tensor_id.second; local
177 if (output_index == Graph::kControlSlot) {
182 if (it == name_to_cost.end() || output_index < 0) {
190 input_cost->output_info(output_index);
|
| /external/valgrind/coregrind/m_gdbserver/ |
| remote-utils.c | 701 int input_index, output_index; local
703 output_index = 0;
709 if (output_index + 2 > out_maxlen)
711 out_buf[output_index++] = '}';
712 out_buf[output_index++] = b ^ 0x20;
714 if (output_index + 1 > out_maxlen)
716 out_buf[output_index++] = b;
721 return output_index;
737 int input_index, output_index; local
740 output_index = 0
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.cc | 403 int output_index) {
407 set_output_ref(output_index, (*params_->inputs)[input_index].mutex_if_ref,
412 int input_index, int output_index, const TensorShape& output_shape,
416 : output_alloc_attr(output_index);
418 input_index, expected_output_dtype(output_index), output_shape,
419 output_memory_type(output_index), output_attr);
422 outputs_[output_index] = TensorValue(new_tensor.release());
423 *output = outputs_[output_index].tensor;
433 int input_index, output_index, stop; local
443 params_->op_kernel->OutputRange(output_name, &output_index, &stop))
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 457 const int output_index = 0; local
458 const DataType dt = node.output_type(output_index);
465 shape_inference::ShapeHandle shape_handle = context->output(output_index);
914 const int output_index = i; local
[all...] |
| /toolchain/binutils/binutils-2.27/bfd/ |
| cofflink.c | 1428 unsigned long output_index; local
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass.cc | 746 int output_index = output_iter->second; local
780 int output_index = output.second; local
822 int output_index = output.second; local
1112 int output_index = output.second; local
[all...] |
