Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc64 int64_t operands_size = get_size(inst->getOperandTypes()); in ShouldBeFolded() local
67 (results_size <= kSizeFactor * operands_size); in ShouldBeFolded()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dwhile_loop_outline.cc244 const int operands_size = while_op.getNumOperands() + extra_operands.size(); in OutlineWhile() local
246 operands.reserve(operands_size); in OutlineWhile()
250 new_types.reserve(operands_size); in OutlineWhile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.cc306 int64 operands_size = 0; in ComputeGloballyUnfusible() local
308 operands_size += total_size(op->shape()); in ComputeGloballyUnfusible()
310 if (operands_size <= total_size(producer->shape())) { in ComputeGloballyUnfusible()