Searched defs:batch_dims_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | gather_op.cc | 190 int32 batch_dims_ = 0; member in tensorflow::GatherOp |
| D | fingerprint_op_test.cc | 60 Tensor batch_dims_; member in tensorflow::__anon380dd9300111::FingerprintOpTest |
| D | resource_variable_ops.cc | 730 int32 batch_dims_ = 0; member in tensorflow::ResourceGatherOp |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| D | gather_op.cc | 261 int32 batch_dims_ = 0; member in tensorflow::GatherOp |
| D | variable_ops.cc | 143 int32 batch_dims_; member in tensorflow::__anon798bfbf40111::ResourceGatherOp |