| /external/tensorflow/tensorflow/core/kernels/ |
| list_kernels.h | 166 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); variable
170 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
174 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
479 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); variable
483 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
487 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
655 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); variable
659 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
663 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
[all...] |
| segment_reduction_ops.cc | 114 auto output_flat = output->flat_outer_dims<T>(); variable
164 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape);
168 T* out_slice_ptr = &output_flat(out_index, 0);
284 auto output_flat = output->flat_outer_dims<T>();
289 output_flat);
493 auto output_flat = output->flat_outer_dims<T>(); variable
496 data.NumElements(), data_ptr, output_flat);
694 auto output_flat = output->flat_outer_dims<T>(); variable
729 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape);
733 auto out = output_flat.template chip<0>(out_index)
1069 auto output_flat = output->flat_outer_dims<T>(); variable
[all...] |
| tensor_array_ops.cc | 685 auto output_flat = variable
706 ConcatGPU<T>(ctx, input_tensors_flat, output_tensor, &output_flat);
710 ConcatCPU<T>(ctx->device(), input_tensors_flat, &output_flat);
934 auto output_flat = variable
[all...] |
| batch_kernels.cc | 84 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); local
87 ConcatGPU<T>(context, inputs_flat, output, &output_flat);
91 ConcatCPU<T>(context->device(), inputs_flat, &output_flat);
[all...] |
