flat_shape - Code Search

tensorflow/c/experimental/saved_model/core/ops/restore_ops.cc

                                      const std::string& s,
                                      ImmediateTensorHandlePtr* out) {
  int64_t flat_shape[] = {1};
  AbstractTensorPtr tensor(ctx->CreateTensor(DT_STRING, flat_shape));
  if (tensor.get() == nullptr) {
    return errors::Internal(
        "Failed to create vector string tensor for checkpoint restore");
  }

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Wed Aug 11 01:20:50 UTC 2021

- 4.5K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/jit/pjrt_tensor_buffer_util_test.cc

  std::vector<int32_t> data{1, 2, 3, 4, 5, 6};
  xla::Shape xla_shape = xla::ShapeUtil::MakeShape(xla::S32, dimensions);
  TF_ASSERT_OK_AND_ASSIGN(
      auto pjrt_buffer,
      pjrt_client->BufferFromHostBuffer(
          data.data(), xla_shape.element_type(), xla_shape.dimensions(),
          /*byte_strides=*/std::nullopt,

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Thu Sep 14 18:14:47 UTC 2023

- 2.7K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/jit/xla_tpu_device.cc

    const TensorShape& shape, DataType type, bool use_fast_memory,
    XlaLayoutPreference layout_preference) {
  xla::Shape xla_shape;
  TF_RETURN_IF_ERROR(
      tensorflow::TensorShapeToXLAShape(type, shape, &xla_shape));
  ApiConverter::StackHelper<XLA_Shape> se_shape(xla_shape);
  ApiConverter::StackHelper<XLA_Shape> tpu_shape;
  StatusHelper status;
  stream_executor::tpu::ExecutorApiFn()->XlaShapeToTpuShapeRepresentationFn(

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Tue May 28 22:53:47 UTC 2024

- 20.9K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc

  mlir::FunctionType func_type = main_fn.getFunctionType();
  for (auto input_type : func_type.getInputs()) {
    tensorflow::TensorShape tensor_shape;
    xla::Shape xla_shape = xla::TypeToShape(input_type);
    TF_RETURN_IF_ERROR(tensorflow::TensorShape::BuildTensorShape(
        xla_shape.dimensions(), &tensor_shape));
    arg_shapes.emplace_back(tensor_shape);

    DataType dtype;
    TF_RETURN_IF_ERROR(ConvertToDataType(input_type, &dtype));

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Thu Jun 13 23:59:33 UTC 2024

- 3.9K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

      [shape_determination_fns](
          const xla::Shape& xla_shape) -> absl::StatusOr<xla::Shape> {
    TensorShape shape;
    TF_RETURN_IF_ERROR(XLAShapeToTensorShape(xla_shape, &shape));
    TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType(
                                            xla_shape.element_type()));
    auto layout_preference = shape_determination_fns.layout_preference_fn(

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Tue May 21 17:24:39 UTC 2024

- 45.3K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc

  // this can fail if we're not running on a TPU-enabled node.
  // TODO(kramm): Move this into a separate pass. See b/184944903
  xla::Shape old_shape = xla::TypeToShape(t);
  XLA_Shape old_shape_c = {};
  XLA_Shape new_shape_c = {};
  TfTpu_ExecutorApiFn *executor = stream_executor::tpu::ExecutorApiFn();
  if (!stream_executor::tpu::IsInitialized(executor)) {
    return failure();
  }

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Thu Apr 25 16:01:03 UTC 2024

- 6.1K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

    TF_RETURN_IF_ERROR(tensorflow::ConvertToTensor(
        llvm::dyn_cast<TF::ConstOp>(fold_results[0].getDefiningOp()).getValue(),
        &tensor));
    xla::Shape xla_shape;
    TF_RETURN_IF_ERROR(tensorflow::TensorShapeToXLAShape(
        tensor.dtype(), tensor.shape(), &xla_shape));
    xla::PjRtClient::HostBufferSemantics host_buffer_semantics =
        xla::PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes;

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Wed Apr 03 01:03:21 UTC 2024

- 35.8K bytes

- Viewed (0)

github.com/golang/go

src/cmd/vendor/rsc.io/markdown/emoji.go

	"five":                                 "5\ufe0f\u20e3",
	"flags":                                "\U0001f38f",
	"flamingo":                             "\U0001f9a9",
	"flashlight":                           "\U0001f526",
	"flat_shoe":                            "\U0001f97f",
	"flatbread":                            "\U0001fad3",
	"fleur_de_lis":                         "\u269c\ufe0f",
	"flight_arrival":                       "\U0001f6ec",

Registered: Wed Jun 12 16:32:35 UTC 2024

- Last Modified: Wed Jan 24 13:01:26 UTC 2024

- 107.7K bytes

- Viewed (0)

Search Options