| /external/tensorflow/tensorflow/core/kernels/ |
| concat_lib_gpu.cc | 37 inputs_flat,
53 inputs_flat,
55 CudaDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size());
57 for (int i = 0; i < inputs_flat.size(); ++i) {
58 input_ptrs.Set(i, inputs_flat[i]->data());
62 CudaDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1);
67 for (int i = 0; i < inputs_flat.size(); ++i) {
68 if (one_size_input && i < inputs_flat.size() - 1 &&
69 inputs_flat[i]->dimension(1) != inputs_flat[i + 1]->dimension(1))
[all...] |
| pack_op.cc | 109 ConstMatrixVector inputs_flat; variable
110 inputs_flat.reserve(num);
112 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
117 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
123 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat);
127 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| concat_lib.h | 54 inputs_flat,
|
| concat_lib_gpu_impl.cu.cc | 118 inputs_flat,
121 for (int i = 0; i < inputs_flat.size(); ++i) {
123 size[0] = inputs_flat[i]->dimension(0);
124 size[1] = inputs_flat[i]->dimension(1);
127 To32Bit(*inputs_flat[i]);
129 output->slice(offset, size).device(gpu_device) = *inputs_flat[i];
177 inputs_flat, \
184 inputs_flat, \
|
| concat_op.cc | 86 ConstMatrixVector inputs_flat;
87 inputs_flat.reserve(N);
116 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
137 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
143 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat);
147 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| quantized_concat_op.cc | 121 ConstMatrixVector* inputs_flat,
128 inputs_flat->reserve(N);
153 inputs_flat->emplace_back(new typename TTypes<T, 2>::ConstMatrix(
201 ConstMatrixVector inputs_flat; variable
204 inputs_flat_dim0, &inputs_flat, &output_concat_dim);
220 context->device(), inputs_flat, sizeof(T) /* cost_per_unit */,
|
| list_kernels.h | 105 ConstMatrixVector inputs_flat; variable
106 inputs_flat.reserve(l->tensors.size());
114 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
121 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
125 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| mkl_concat_op.cc | 99 ConstMatrixVector inputs_flat;
100 inputs_flat.reserve(N);
129 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
148 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
187 ConstMatrixVector inputs_flat;
188 inputs_flat.reserve(N);
208 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
226 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
[all...] |
| batch_kernels.cc | 51 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; local
52 inputs_flat.reserve(inputs.size());
71 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
86 ConcatGPU<T>(context, inputs_flat, output, &output_flat);
90 ConcatCPU<T>(context->device(), inputs_flat, &output_flat);
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| constant_op.py | 63 inputs_flat = [tensor, shape]
66 b"Reshape", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
74 inputs_flat = [dims, value]
77 b"Fill", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
|
