Searched defs:scatter_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | scatter_spmd_expander.cc | 77 auto scatter_op = llvm::cast<OpType>(op); in TensorScatterOpExpand() local
157 auto scatter_op = llvm::cast<OpType>(op); in TensorScatterOpComputeLayoutForward() local
185 auto scatter_op = llvm::cast<OpType>(op); in TensorScatterOpComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | embedding_ops_test.py | 51 def _TestCase(self, shape, indices, scatter_op=state_ops.scatter_add): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 810 auto scatter_op = func_builder->create<mlir::mhlo::ScatterOp>( in ImportInstructionImpl() local
|
| D | mlir_hlo_to_hlo.cc | 1490 auto scatter_op = xla::Scatter(operands, scatter_indices, updates, in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 2799 Operation *scatter_op, llvm::ArrayRef<int64_t> update_window_dims, in CanonicalizeScatterUpdates()
3132 mhlo::ScatterOp scatter_op, OpAdaptor adaptor, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2207 auto scatter_op = mlir::cast<mlir::lmhlo::ScatterOp>(op); in EmitScatter() local
|