- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for scalar_tensor (0.1 sec)
-
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) -
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)