Searched refs:data_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | bias_op_deterministic_test.py | 42 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument
44 data_dims = data_rank * (data_dim,)
91 data_dim = 14
93 data_dim, data_layout)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_gpu.cu.cc | 260 int32 data_dim, T* var, T* m, T* v, const T* const beta1_power_, in ApplyAdamKernel() argument
277 for (int32 i = blockIdx.x * blockDim.x + threadIdx.x; i < data_dim; in ApplyAdamKernel()
448 int32 data_dim = var.dimension(0); in wrap_kernel_call() local
449 auto config = GetGpuLaunchConfig(data_dim, d); in wrap_kernel_call()
885 int32 data_dim = grad.dimension(0); in operator ()() local
886 GpuLaunchConfig config = GetGpuLaunchConfig(data_dim, d); in operator ()()
894 d.stream(), data_dim, var.data(), m.data(), v.data(), in operator ()()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2729 int64_t data_dim = std::get<1>(shape_pair); in VerifyUnsortedSegmentReduction() local
2731 !ShapedType::isDynamic(data_dim) && segment_id_dim != data_dim) in VerifyUnsortedSegmentReduction()
2736 << data_dim; in VerifyUnsortedSegmentReduction()
|