Searched refs:VariableOp (Results 1 – 9 of 9) sorted by relevance
31 class VariableOp : public OpKernel {33 explicit VariableOp(OpKernelConstruction* context);41 TF_DISALLOW_COPY_AND_ASSIGN(VariableOp);
68 VariableOp::VariableOp(OpKernelConstruction* context) : OpKernel(context) { in VariableOp() function in tensorflow::VariableOp75 void VariableOp::Compute(OpKernelContext* ctx) { in Compute()194 REGISTER_KERNEL_BUILDER(Name("Variable").Device(DEVICE_CPU), VariableOp);195 REGISTER_KERNEL_BUILDER(Name("VariableV2").Device(DEVICE_CPU), VariableOp);210 VariableOp); \213 VariableOp); \
26 DenseMap<Operation *, SmallVector<spirv::VariableOp, 8>> outputVars;107 auto alloc = rewriter.create<spirv::VariableOp>( in replaceSCFOutputValue()
1242 class VariablePattern : public SPIRVToLLVMConversion<spirv::VariableOp> {1244 using SPIRVToLLVMConversion<spirv::VariableOp>::SPIRVToLLVMConversion;1247 matchAndRewrite(spirv::VariableOp varOp, ArrayRef<Value> operands, in matchAndRewrite()
219 LogicalResult processVariableOp(spirv::VariableOp op);906 LogicalResult Serializer::processVariableOp(spirv::VariableOp op) { in processVariableOp()1925 .Case([&](spirv::VariableOp op) { return processVariableOp(op); }) in processOperation()
676 friend class VariableOp; // For access to set_shape. variable
521 // Conversion of VariableOp without initialization525 // Conversion of VariableOp with initialization
2930 static void print(spirv::VariableOp varOp, OpAsmPrinter &printer) { in print()2933 printer << spirv::VariableOp::getOperationName(); in print()2943 static LogicalResult verify(spirv::VariableOp varOp) { in verify()
2794 void VariableOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()