- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 453 for int64Opt (0.14 sec)
-
tensorflow/c/experimental/gradients/grad_test_helper.cc
} } void CheckTensorValue(AbstractTensorHandle* t, absl::Span<const float> manuals, absl::Span<const int64_t> dims, double abs_error) { TF_Tensor* analytical_tensor; auto s = GetValue(t, &analytical_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); int64_t num_elem_analytical = 1; auto num_dims_analytical = TF_NumDims(analytical_tensor); ASSERT_EQ(dims.size(), num_dims_analytical);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/types.cc
.def("get", &mlir::UnrankedTensorType::get); py::class_<mlir::RankedTensorType, mlir::Type>(m, "RankedTensorType") .def("get", [](std::vector<int64_t> shape, mlir::Type ty) { return mlir::RankedTensorType::get(mlir::ArrayRef<int64_t>(shape), ty); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 12 08:42:26 UTC 2021 - 2K bytes - Viewed (0) -
tensorflow/c/ops.h
TF_ShapeHandle* result, TF_Status* status); // Places a handle to the ith dimension of the given shape into *result. TF_CAPI_EXPORT extern void TF_ShapeInferenceContextDim( TF_ShapeInferenceContext* ctx, TF_ShapeHandle* shape_handle, int64_t i, TF_DimensionHandle* result); // Returns in <*result> a sub-shape of <shape_handle>, with dimensions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 16.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.cc
ResourceHandleValueAndId GetResourceHandleValueAndIdBase( llvm::StringRef container, llvm::StringRef shared_name, llvm::StringRef device, Value resource, llvm::SmallDenseMap<ResourceHandle, int64_t> &resource_handle_id_map, int64_t &next_id) { // Always create a new ID for anonymous handle. if (IsResourceHandleAnonymous(shared_name)) return {resource, next_id++};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 03 15:22:46 UTC 2021 - 1.8K bytes - Viewed (0) -
tensorflow/c/eager/abstract_operation.h
virtual Status SetAttrInt(const char* attr_name, int64_t value) = 0; virtual Status SetAttrFloat(const char* attr_name, float value) = 0; virtual Status SetAttrBool(const char* attr_name, bool value) = 0; virtual Status SetAttrType(const char* attr_name, DataType value) = 0; virtual Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 14 16:20:41 UTC 2021 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc
AddInputFromArray<float>(TensorShape({}), {1.f}); AddInputFromArray<float>(TensorShape({}), {5.f}); AddInputFromArray<int64_t>(TensorShape({0}), {}); AddInputFromArray<float>(TensorShape({}), {1.f}); AddInputFromArray<float>(TensorShape({}), {5.f}); AddInputFromArray<int64_t>(TensorShape({8}), {1, 4, 6, 7, 3, 2, 1, 0}); TF_CHECK_OK(RunOpKernel()); kernel_.reset();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
typedef StablehloConvolutionOptions TableType; std::vector<int64_t> window_strides{}; std::vector<int64_t> padding{}; std::vector<int64_t> lhs_dilation{}; std::vector<int64_t> rhs_dilation{}; std::vector<bool> window_reversal{}; int64_t input_batch_dimension = 0; int64_t input_feature_dimension = 0; std::vector<int64_t> input_spatial_dimensions{}; int64_t kernel_input_feature_dimension = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/test_resource_alias_analysis.cc
const TF::ResourceAliasAnalysis::Info& analysis) { int64_t next_id = 0; llvm::SmallDenseMap<Value, int64_t, 8> ids; auto assign_id = [&](Value value) { if (ids.find(value) == ids.end()) ids.insert({value, next_id++}); }; auto get_id = [&](Value value) -> int64_t { auto it = ids.find(value); assert(it != ids.end()); return it->second; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
Output slice = ops::Slice(root.WithOpName("slice"), input, begin, size); std::unique_ptr<Graph> result; TF_ASSERT_OK(IncreaseDynamismForAutoJit(root, &result)); const int64_t zero_64 = 0; const int32_t zero_32 = 0; const int64_t one_64 = 1; auto m_input = Out(NodeWith(Op("Placeholder"), Name("input"))); auto m_begin_s64 = Out(NodeWith( Op("Cast"), Inputs(Out(NodeWith(Op("Placeholder"), Name("begin"))))));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/hardwares/cpu_hardware.cc
// tfl.conv_2d / tfl.depthwise_conv_2d / tfl.fully_connected class CpuConvOp : public TargetHardwareOperation { double GetOpCost(mlir::Operation* op) const override { float cost = 0.0; int64_t arithmetic_count; if (ArithmeticCountUtilHelper::GetArithmeticCountForConvAndFullyconnectedOp( op, &arithmetic_count)) { cost = arithmetic_count * kCPUArithmeticUnitCost; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 5.9K bytes - Viewed (0)