| /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 | pack_op.cc | 109 ConstMatrixVector inputs_flat; in Compute() local
110 inputs_flat.reserve(num); in Compute()
112 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
117 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
123 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
127 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| 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()
178 inputs_flat, \
185 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()
201 ConstMatrixVector inputs_flat; in Compute() local
204 inputs_flat_dim0, &inputs_flat, &output_concat_dim); in Compute()
220 context->device(), inputs_flat, sizeof(T) /* cost_per_unit */, in Compute()
|
| D | list_kernels.h | 143 ConstMatrixVector inputs_flat; in Compute() local
144 inputs_flat.reserve(tensor_list->tensors.size()); in Compute()
148 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
161 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
170 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
174 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
451 ConstMatrixVector inputs_flat; in Compute() local
452 inputs_flat.reserve(tensor_list->tensors.size()); in Compute()
459 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
474 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
[all …]
|
| D | concat_lib_gpu.h | 36 inputs_flat,
51 inputs_flat, \
56 inputs_flat, \
|
| D | concat_op.cc | 108 ConstMatrixVector inputs_flat; in Compute() local
109 inputs_flat.reserve(N); in Compute()
138 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
159 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
165 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
169 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | concat_lib.h | 55 inputs_flat,
63 inputs_flat, \
|
| D | mkl_concat_op.cc | 95 ConstMatrixVector inputs_flat; in Compute() local
96 inputs_flat.reserve(N); in Compute()
116 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Compute()
134 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | batch_kernels.cc | 53 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; in Concat() local
54 inputs_flat.reserve(inputs.size()); in Concat()
73 inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( in Concat()
87 ConcatGPU<T>(context, inputs_flat, output, &output_flat); in Concat()
91 ConcatCPU<T>(context->device(), inputs_flat, &output_flat); in Concat()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op.py | 45 inputs_flat = [tensor, shape]
48 b"Reshape", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
56 inputs_flat = [dims, value]
59 b"Fill", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx)
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 289 inputs_flat = nest.flatten(inputs)
290 batch_size = array_ops.shape(inputs_flat[0])[1]
313 inputs_flat[0].shape[0],
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 696 inputs_flat = K.reshape(inputs, (K.shape(inputs)[0], -1))
701 output_flat = K.math_ops.sparse_matmul(inputs_flat, kernel, b_is_sparse=True)
|