Home
last modified time | relevance | path

Searched defs:batch_dims_ (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgather_op.cc190 int32 batch_dims_ = 0; member in tensorflow::GatherOp
H A Dfingerprint_op_test.cc60 Tensor batch_dims_; member in tensorflow::__anonac4a473f0111::FingerprintOpTest
H A Dresource_variable_ops.cc801 int32 batch_dims_ = 0; member in tensorflow::ResourceGatherOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc265 int32 batch_dims_ = 0; member in tensorflow::GatherOp
H A Dvariable_ops.cc159 int32 batch_dims_; member in tensorflow::__anon735ed9830111::ResourceGatherOp