Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc69 gemm_config.set_batch_size(batch_size); in HandleDot()
Dir_emitter_unnested.cc1236 backend.set_batch_size(op.batch_size()); in EmitGemmThunkFromMlir()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h313 void set_batch_size(int batch_size) { batch_size_ = batch_size; } in set_batch_size() function
Dconvert_nodes.cc1564 input.set_batch_size(batch_size_); in AddTensorOrWeights()