Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc401 const mlir::Value zero_scalar = ops_util::CreateScalarConst(0, builder, loc); in CreateZeroScalar() local
481 const mlir::Value zero_scalar = CreateZeroScalar(builder, loc, input_type); in LowerAllGatherOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_sparsetensors.cc211 auto zero_scalar = CreateZeroScalarConst(builder, front_op->getLoc(), in runOnOperation() local
H A Ddtensor_allreduce_combine_optimization.cc183 auto zero_scalar = ops_util::CreateScalarConst(0, builder, loc); in MergeAllReduceGroup() local
H A Dcollectives.cc177 auto zero_scalar = CreateZeroScalarConst( in EmitDenseToSparseToDense() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_group_converter.cc371 auto zero_scalar = in HandleBatchGroupCount() local