Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 115 for getBlob (0.15 sec)

  1. tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc

                builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(),
                                               cst.getValue(), s_param, new_value);
            value.replaceAllUsesWith(s_qconst.getResult());
            cst.erase();
          }
    
          if (result.needs_densify) {
            auto value = op->getOperand(operand);
            auto densify =
                builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

      for (int64_t dim : element_shape) {
        buffer_shape.push_back(dim);
      }
      auto zero = CreateScalarConst(0, builder, op->getLoc());
      if (getElementTypeOrSelf(zero.getType()) != element_dtype) {
        zero = builder.create<TF::CastOp>(
            op->getLoc(),
            ArrayRef<Type>{tensorflow::GetTypeFromTFTensorShape({}, element_dtype)},
            ArrayRef<Value>{zero});
      }
      auto buffer_type =
    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/lite/transforms/legalize_tensorlist.cc

        // It is currently not possible to easily pack the output of a multi-result
        // op into an op with a single varidic output in `.td`.
        auto converted = rewriter.create<TFL::CustomOp>(
            op->getLoc(), op->getResultTypes(), op->getOperands(),
            "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), ""));
        rewriter.replaceOp(op, converted.getResults());
        return success();
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td

        "$_builder.create<TF::ShapeOp>($0.getLoc(), $1, $2)">;
    
    def IsI32 : NativeCodeCall<
        "$_builder.getBoolAttr(getElementTypeOrSelf($0.getType()).isInteger(32))">;
    
    def CreateTFCastOpF32 : NativeCodeCall<
        "CreateTFCastOpF32(&$_builder, $0.getLoc(), $1, $2)">;
    
    def CreateTFCastOpI32 : NativeCodeCall<
        "CreateTFCastOpI32(&$_builder, $0.getLoc(), $1, $2)">;
    
    def CreateTensorScatterNdOp : NativeCodeCall<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 24.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

                              op.getLoc(), &rewriter);
      } else if (out_reshape_need) {
        out = createOutputReshapeOpForDynamic(out, reshape_shape, original_lhs,
                                              original_rhs, original_dnums,
                                              op.getLoc(), &rewriter);
      }
      out = createTransposeOp(out, op.getLoc(), out_transpose, &rewriter);
    
      rewriter.replaceOp(op, out);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

        for (const auto& arg : func_op.getArguments()) {
          arg_locs.push_back(arg.getLoc());
        }
      }
    
      // Creates a new main function.
      auto func_type = FunctionType::get(context, arg_types, result_types);
      auto main_func = builder.create<func::FuncOp>(
          module_op.getLoc(), kImportModelDefaultGraphFuncName, func_type);
      builder.createBlock(&main_func.getBody(), main_func.begin(), arg_types,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc

        }
      }
      auto island = OpBuilder(fetch).create<IslandOp>(
          fetch.getLoc(), data_types,
          /*control=*/ControlType::get(fetch.getContext()),
          /*controlInputs=*/control_fetches);
      island.getBody().push_back(new Block);
      OpBuilder::atBlockEnd(&island.GetBody())
          .create<YieldOp>(fetch.getLoc(), data_fetches);
      const int fetch_control_idx = data_fetches.size();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

            // and once existing cases have been fixed.
            mlir::InFlightDiagnostic error =
                strict_clusters ? mlir::emitError(op.getLoc(), "")
                                : mlir::emitWarning(op.getLoc(), "");
            error << "Op has cyclic dependency with a compilation cluster:\n";
            error << "The cluster depends on\n";
            error << op.getName() << "\n"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc

        return failure();
      }
    
      auto tfl_fusable_op = builder.create<TFL::CustomOp>(
          func->getLoc(), func.getFunctionType().getResults(), func.getArguments(),
          custom_op_name, CustomOption(&builder, custom_option_buffer));
      builder.create<func::ReturnOp>(func->getLoc(), tfl_fusable_op.getResults());
      return success();
    }
    
    // Abstracts the conversion of the embedded lookup composite function.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc

      auto merged_execute_launch = builder->create<tf_device::LaunchOp>(
          merged_execute.getLoc(), execute_launch.getDeviceAttr(),
          merged_execute.getResultTypes());
      merged_execute_launch.getBody().push_back(new Block);
    
      builder->setInsertionPointToEnd(&merged_execute_launch.GetBody());
      builder->create<tf_device::ReturnOp>(merged_execute.getLoc(),
                                           merged_execute.getResults());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 17:52:11 UTC 2024
    - 27K bytes
    - Viewed (0)
Back to top