Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for GetOperation (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

        mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>();
        auto symbol =
            mlir::SymbolTable::lookupNearestSymbolFrom(call.getOperation(), sym);
        if (!symbol) return failure();
        auto f = llvm::dyn_cast<mlir::func::FuncOp>(symbol);
    
        if (call.getOperation()->getNumOperands() !=
            f.getBody().front().getNumArguments()) {
          return failure();  // RemoteCall et al
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc

      // The passes to run on the module.
      std::vector<GraphOptimizationPass*> passes_;
    };
    }  // anonymous namespace
    
    void GraphOptPass::runOnOperation() {
      mlir::ModuleOp module_in = getOperation();
      mlir::MLIRContext& ctx = getContext();
    
      // Convert MLIR to Graph
      FunctionLibraryDefinition flib_def(OpRegistry::Global(),
                                         FunctionDefLibrary());
      GraphExportConfig confs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        if (output_shape.value()) {
          changed = RefineResultType(op.getOperation(), op.getResult(),
                                     output_shape.value());
          return changed;
        }
      }
      return changed;
    }
    
    bool ShapeInference::RefineWithInferTypeOpInterface(
        InferTypeOpInterface infer_ti) {
      Operation* op = infer_ti.getOperation();
      if (none_of(op->getResultTypes(), CanBeRefined)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        ParseResult $cppClass::parse(OpAsmParser &parser, OperationState &result) {
          return parseOneResultSameOperandTypeOp(parser, result);
        }
        void $cppClass::print(OpAsmPrinter &p) {
          return printOneResultOp(getOperation(), p);
        }
      }];
    
      let hasOptions = 1;
    }
    
    def TFL_AddNOp : TFL_Op<"add_n", [
        Commutative,
        Pure,
        SameOperandsAndResultsScale,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc

            operand.assign(orig_to_new[operand.get()]);
          }
        }
        return WalkResult::advance();
      });
    
      return success();
    }
    
    void XlaBroadcast::runOnOperation() {
      FuncOp func = getOperation();
      mlir::ModuleOp module = func->getParentOfType<mlir::ModuleOp>();
      if (!module) return signalPassFailure();
      func.walk([&](ClusterOp cluster) {
        if (auto replicate = cluster->getParentOfType<ReplicateOp>()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 18:52:07 UTC 2024
    - 13.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

                                       const std::vector<int32_t>& operands,
                                       const std::vector<int32_t>& results) {
      std::string op_name =
          gather_op.getOperation()->getName().getStringRef().str();
      uint32_t opcode_index =
          GetOpcodeIndex(op_name, tflite::BuiltinOperator_STABLEHLO_GATHER);
    
      std::vector<int64_t> offset_dims_vec(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        }
        return WalkResult::advance();
      });
      return res.wasInterrupted();
    }
    
    void LowerStaticTensorListPass::runOnOperation() {
      auto *context = &getContext();
      ModuleOp module = getOperation();
    
      // When allow_tensorlist_pass_through_ == false the target dynamic legal
      // checks will cause this pass to fail if there are any variant type tensors
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

            stride_values);
    
        auto d_slice = rewriter.create<mhlo::RealDynamicSliceOp>(
            loc, op.getOperation()->getResult(0).getType(), input, start_indices,
            end_indices, stride_indices);
        rewriter.replaceOp(op, d_slice.getOperation()->getResults());
        return success();
      }
    };
    
    static void BroadcastBatchMatMulV2Operands(Value lhs, Value rhs, Location loc,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

          // has two elements, so it will be merging two operand ranges (the two
          // variadic arguments). Attributes like `batch_timeout_micros` will not
          // make it to the return array.
          auto segment_sizes = getOperation()->getAttrOfType<DenseI32ArrayAttr>(getOperandSegmentSizeAttr());
          return OperandRangeRange(getOperands(), segment_sizes).join();
        }
      }];
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top