Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 14 of 14 for ConvertTensor (0.66 sec)

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

        if (tensors.empty()) {
          tensorflow::Tensor tensor(list->element_dtype,
                                    tensorflow::TensorShape(result_shape));
          auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter);
          if (!attr_or.ok()) return failure();
          rewriter.replaceOpWithNewOp<TF::ConstOp>(op, attr_or.value());
          return success();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

                          xla_literal->shape().element_type()) *
                          xla_literal->element_count());
          TF_ASSIGN_OR_RETURN(auto attrs,
                              tensorflow::ConvertTensor(tensor, &builder));
          builder.setInsertionPoint(
              &func_op.getFunctionBody().getBlocks().front().front());
          // Use mhlo.Constant when it is consumed by the lowering passes since they
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 01:03:21 UTC 2024
    - 35.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        }
        if (xla_arg.kind != XlaArgument::kConstant) {
          params.push_back(idx);
          continue;
        }
    
        TF_ASSIGN_OR_RETURN(auto value_attr,
                            ConvertTensor(xla_arg.constant_value, &builder));
        // TODO(hinsu): Use the actual location of the constant.
        auto constant = builder.create<mlir::TF::ConstOp>(
            mlir::UnknownLoc::get(module_op.getContext()), value_attr);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                saved_model->variable_reader()->Lookup(checkpoint_key, &value),
                "Could not read checkpoint key from variables bundle: ",
                checkpoint_key);
            TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(value, &builder));
            // A variable can have a partially known type, such as
            // tensor<?x27x?xf32>, even if the initializer is a specific static
            // shape.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top