- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for scalar (0.17 sec)
-
tensorflow/c/eager/gradient_checker.cc
int num_dims_out = TF_NumDims(model_out_tensor); TF_DeleteTensor(model_out_tensor); // If the output is a scalar, then return the scalar output if (num_dims_out == 0) { outputs[0] = model_out.release(); return absl::OkStatus(); } // Else, reduce sum the output to get a scalar // Will sum all dimensions, so get a Tensor containing [0,...,num_dims_out-1]. AbstractTensorHandlePtr sum_dims; {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
} public: bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } bool UseFunction() const { return std::get<2>(GetParam()); } }; // Checks that inputs[0] is a scalar. Status TestScalarShape(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { PartialTensorShape shape;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0)