Searched defs:TensorScatterElements (Results 1 – 4 of 4) sorted by relevance
35 TensorScatterElements() : BaseOperator(kNameTensorScatterElements) { in TensorScatterElements() function
58 cudaError_t TensorScatterElements(const enum TensorScatterElementsReductionType reduction_type, con… in TensorScatterElements() function
565 inline NodePtr TensorScatterElements(Emitter *ib, const NodePtr &input, const ValuePtr &dim, const … in TensorScatterElements() function
4965 class TensorScatterElements(Primitive): class