Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for scalar_tensor (0.1 sec)

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

      // Just create a constant to replace its uses.
      tensorflow::Tensor scalar_tensor(tensorflow::DT_FLOAT, {});
      scalar_tensor.scalar<float>()() = 0.0f;
      auto flow = builder.create<TF::ConstOp>(
          ta.getLoc(), tensorflow::ConvertTensor(scalar_tensor, &builder).value());
      ta.getFlow().replaceAllUsesWith(flow);
      ta.erase();
      (*stats)[local_var].accumulate_on_write = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental_test.cc

      TFE_TensorHandle* scalar = TestScalarTensorHandle(tfe_context_, five);
      TF_Tensor* scalarTensor = TFE_TensorHandleResolve(scalar, status_);
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      CheckOutputShapes(fill_op,
                        /* input_shapes*/ {unknown_shape(), unknown_shape()},
                        /* input_tensors*/ {tensor_1X1X6, scalarTensor},
                        /*expected_shape*/ make_shape({1, 1, 6}));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
Back to top