Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Drewrite_tpu_embedding_ops.cc90 int32_t operand_sizes[] = {static_cast<int32_t>(send_op.N()), in runOnFunction() local
93 auto operand_size_attr = DenseIntElementsAttr::get(attr_ty, operand_sizes); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc293 SmallVector<Value, 2> operand_sizes(operand_rank, one); in InsertDynamicMemrefCastOp() local
300 operand_sizes[i] = operand_dim_size; in InsertDynamicMemrefCastOp()
342 loc, CmpIPredicate::slt, operand_sizes[dim], result_dim_size); in InsertDynamicMemrefCastOp()