Searched defs:data_dim (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | bias_op_d9m_test.py | 37 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops_gpu.cu.cc | 182 int32 data_dim, T* var, T* m, T* v, const T* const beta1_power_, in ApplyAdamKernel()
375 int32 data_dim = var.dimension(0); in wrap_kernel_call() local
875 int32 data_dim = grad.dimension(0); in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 2923 int64_t data_dim = std::get<1>(shape_pair); in VerifyUnsortedSegmentReduction() local
|