Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ScatterAccumulateElements (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.h

                         Location loc);
    
    // Scatters elements into buffer, where each scattered element is accumulated
    // with the old value in buffer.
    Value ScatterAccumulateElements(Value indices, Value updates, Value buffer,
                                    OpBuilder builder, Location loc);
    
    }  // namespace collection_ops_util
    }  // namespace TF
    }  // namespace mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 4.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

      return builder.create<TF::GatherV2Op>(
          loc, ArrayRef<Type>{result_type},
          ArrayRef<Value>{buffer, indices, CreateScalarConst(0, builder, loc)});
    }
    
    Value ScatterAccumulateElements(Value indices, Value updates, Value buffer,
                                    OpBuilder builder, Location loc) {
      auto buffer_type = mlir::cast<RankedTensorType>(buffer.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc

        return scatter.emitOpError("unknown tensor array");
      }
      OpBuilder builder(scatter);
      auto buffer = cutil::ReadLocalVariable(local_var, builder, scatter.getLoc());
      buffer =
          cutil::ScatterAccumulateElements(scatter.getIndices(), scatter.getValue(),
                                           buffer, builder, scatter.getLoc());
      cutil::WriteLocalVariable(local_var, buffer, builder, scatter.getLoc());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
Back to top