Home
last modified time | relevance | path

Searched refs:AutoTransposeAndLinearize (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dinfeed_ops.cc195 Status AutoTransposeAndLinearize(OpKernelContext* ctx, in AutoTransposeAndLinearize() function
268 AutoTransposeAndLinearize(ctx, input_tensor, xla_shape_, in Compute()
344 auto status = AutoTransposeAndLinearize( in Compute()