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::__anon623ac64e0111::FingerprintOpTest |
| D | resource_variable_ops.cc | 801 int32 batch_dims_ = 0; member in tensorflow::ResourceGatherOp |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| D | gather_op.cc | 265 int32 batch_dims_ = 0; member in tensorflow::GatherOp |
| D | variable_ops.cc | 159 int32 batch_dims_; member in tensorflow::__anon9c8648120111::ResourceGatherOp |