| /external/tensorflow/tensorflow/core/kernels/ |
| scatter_functor_gpu.cu.h | 35 Index first_dim_size, Index updates_size,
42 if (!(param_first_index >= 0 && param_first_index < first_dim_size)) {
83 const Index first_dim_size = params.dimension(0); local
89 params.data(), updates.data(), indices.data(), first_dim_size,
|
| dynamic_stitch_op_gpu.cu.cc | 53 const int32 slice_size, const int32 first_dim_size,
57 const int32 output_size = first_dim_size * slice_size;
68 const int32 first_dim_size, \
|
| dynamic_stitch_op.cc | 77 OpInputList* data_inputs, int* first_dim_size,
98 *first_dim_size = max_index + 1;
126 // [*first_dim_size] + data.shape[indices.dims:]
128 result_shape.AddDim(*first_dim_size);
140 const int32 slice_size, const int32 first_dim_size,
154 int first_dim_size; variable
158 &first_dim_size, &data_elements_size,
168 if (first_dim_size > 0) {
176 CudaDeviceArrayOnHost<int32> indices_flat(c, first_dim_size);
181 for (int i = 0; i < first_dim_size; ++i)
226 int first_dim_size; variable
[all...] |
| training_ops.cc | 764 const Tindex first_dim_size = var.dim_size(0); local
769 OP_REQUIRES(ctx, index >= 0 && index < first_dim_size,
956 const Tindex first_dim_size = var.dim_size(0); variable
996 const Tindex first_dim_size = var_flat.size(); variable
1297 const Tindex first_dim_size = var.dim_size(0); variable
1324 const Tindex first_dim_size = accum_flat.size(); variable
1448 const Tindex first_dim_size = var.dim_size(0); variable
1491 const Tindex first_dim_size = accum_flat.size(); variable
1742 const Tindex first_dim_size = var.dim_size(0); variable
1788 const Tindex first_dim_size = var_flat.size(); variable
2101 const Tindex first_dim_size = var.dim_size(0); variable
2176 const Tindex first_dim_size = accum_flat.size(); variable
2423 const Tindex first_dim_size = var.dim_size(0); variable
3062 const Tindex first_dim_size = var.dim_size(0); variable
3199 const Tindex first_dim_size = var.dim_size(0); variable
[all...] |
