Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for scalar (0.16 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      const std::vector<int64_t>* shape;
      TF_ASSERT_OK(unknown_length_vector->Shape(&shape));
      EXPECT_THAT(*shape, ElementsAre(-1));
    
      TensorHandlePtr scalar = FloatTensorHandle(2., status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      two_vector = VectorFloatTensorHandle({3., 4.}, status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_test_util.cc

          found_shape = true;
        }
      }
      return found_dtype && found_shape;
    }
    
    bool IsScalarConst(const tensorflow::NodeDef& node_def, int v) {
      if (node_def.op() != "Const" || node_def.name() != "scalar") {
        return false;
      }
      bool found_dtype = false;
      bool found_value = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "dtype") {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  3. tensorflow/c/c_api_experimental_test.cc

      TFE_OpSetAttrType(fill_op, "T", TF_FLOAT);
      TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32);
    
      float five = 5.0;
      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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jan 17 22:27:52 GMT 2023
    - 13.1K bytes
    - Viewed (1)
  4. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

        ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
        delete to_delete;
      };
      std::unique_ptr<Variable, decltype(variable_deleter)> variable(
          Variable::Create(context, TF_FLOAT, /* Scalar */ {}, 0, device_name,
                           status.get()),
          variable_deleter);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
Back to top