| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu.cc | 39 inputs_flat, in ConcatGPUCall()
41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size()); in ConcatGPUCall()
43 for (int i = 0; i < inputs_flat.size(); ++i) { in ConcatGPUCall()
44 input_ptrs.Set(i, inputs_flat[i]->data()); in ConcatGPUCall()
48 GpuDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1); in ConcatGPUCall()
53 for (int i = 0; i < inputs_flat.size(); ++i) { in ConcatGPUCall()
54 if (one_size_input && i < inputs_flat.size() - 1 && in ConcatGPUCall()
55 inputs_flat[i]->dimension(1) != inputs_flat[i + 1]->dimension(1)) { in ConcatGPUCall()
58 scan += inputs_flat[i]->dimension(1); in ConcatGPUCall()
65 inputs_flat[0]->dimension(1), output_flat); in ConcatGPUCall()
[all …]
|
| D | concat_lib_gpu_impl.cu.cc | 119 inputs_flat, in ConcatGPUSlice()
122 for (int i = 0; i < inputs_flat.size(); ++i) { in ConcatGPUSlice()
124 size[0] = inputs_flat[i]->dimension(0); in ConcatGPUSlice()
125 size[1] = inputs_flat[i]->dimension(1); in ConcatGPUSlice()
128 To32Bit(*inputs_flat[i]); in ConcatGPUSlice()
130 output->slice(offset, size).device(gpu_device) = *inputs_flat[i]; in ConcatGPUSlice()
180 inputs_flat, \
187 inputs_flat, \
|
| D | pack_op.cc | 96 ConstMatrixVector inputs_flat; in Compute() local
97 inputs_flat.reserve(num); in Compute()
106 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
111 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
115 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | concat_lib_gpu.h | 36 inputs_flat,
51 inputs_flat, \
56 inputs_flat, \
|
| D | quantized_concat_op.cc | 121 ConstMatrixVector* inputs_flat, in CalculateConcatDims() argument
128 inputs_flat->reserve(N); in CalculateConcatDims()
153 inputs_flat->emplace_back(new typename TTypes<T, 2>::ConstMatrix( in CalculateConcatDims()
199 ConstMatrixVector inputs_flat; in Compute() local
202 inputs_flat_dim0, &inputs_flat, &output_concat_dim); in Compute()
218 context->device(), inputs_flat, sizeof(T) /* cost_per_unit */, in Compute()
|
| D | list_kernels.h | 122 ConstMatrixVector inputs_flat; in Compute() local
123 inputs_flat.reserve(tensor_list->tensors().size()); in Compute()
127 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
140 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
149 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
153 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
426 ConstMatrixVector inputs_flat; in Compute() local
427 inputs_flat.reserve(tensor_list->tensors().size()); in Compute()
435 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
451 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
[all …]
|
| D | concat_lib.h | 56 inputs_flat,
64 inputs_flat, \
|
| D | concat_op.cc | 115 ConstMatrixVector inputs_flat; in Compute() local
116 inputs_flat.reserve(N); in Compute()
143 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
164 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
168 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | concat_split_util.h | 46 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; in Concat() local
47 inputs_flat.reserve(inputs.size()); in Concat()
66 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Concat()
83 ConcatGPU<T>(context, inputs_flat, output, &output_flat); in Concat()
87 ConcatCPU<T>(context->device(), inputs_flat, &output_flat); in Concat()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op.py | 44 inputs_flat = [tensor, shape]
47 b"Reshape", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
55 inputs_flat = [dims, value]
58 b"Fill", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 732 inputs_flat = K.reshape(inputs, (K.shape(inputs)[0], -1))
737 output_flat = math_ops.sparse_matmul(inputs_flat, kernel, b_is_sparse=True)
770 inputs_flat = K.reshape(inputs, (K.shape(inputs)[0], -1))
775 b=inputs_flat,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 174 ConstMatrixVector inputs_flat; in Compute() local
175 inputs_flat.reserve(N); in Compute()
195 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
214 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
217 c->device(), inputs_flat, sizeof(T) /* cost_per_unit */, in Compute()
|