- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for DT_INT64 (0.16 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
.WithXlaCluster("cluster_0"); Output input = ops::Placeholder(root.WithOpName("input"), DT_FLOAT); Output begin = ops::Placeholder(root.WithOpName("begin"), DT_INT64); Output size = ops::Placeholder(root.WithOpName("size"), DT_INT64); Output slice = ops::Slice(root.WithOpName("slice"), input, begin, size); std::unique_ptr<Graph> result; TF_ASSERT_OK(IncreaseDynamismForAutoJit(root, &result));
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/tf2xla/internal/mlir_bridge_pass_util_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
return {slice_inputs}; } // Casts `x` to a DT_INT64 if it isn't one already. Output MakeInt64(const Scope& host_scope, absl::string_view name, const Output& x) { return x.type() == DT_INT64 ? x : ops::Cast(host_scope.WithOpName(name, "_s64"), x, DT_INT64); } // Returns `slice_inputs` with the index and size inputs cast to DT_INT64.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc
inputs.emplace_back("min", 0, DT_FLOAT); inputs.emplace_back("max", 0, DT_FLOAT); inputs.emplace_back("histogram", 0, DT_INT64); inputs.emplace_back("min", 0, DT_FLOAT); inputs.emplace_back("max", 0, DT_FLOAT); inputs.emplace_back("histogram", 0, DT_INT64); const std::string dir = testing::TmpDir(); const std::string output_file_path = io::JoinPath(dir, "statistics.pbtxt");
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/tensorflow/utils/convert_tensor.cc
CONVERT_FLAT(DT_FLOAT, float) CONVERT_FLAT(DT_DOUBLE, double) CONVERT_FLAT(DT_INT8, int8) CONVERT_FLAT(DT_INT16, int16) CONVERT_FLAT(DT_INT32, int32) CONVERT_FLAT(DT_INT64, int64_t) CONVERT_FLAT(DT_UINT8, uint8) CONVERT_FLAT(DT_UINT16, uint16) CONVERT_FLAT(DT_UINT32, uint32) CONVERT_FLAT(DT_UINT64, uint64) CONVERT_FLAT(DT_COMPLEX64, std::complex<float>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
{"y: T"}, // Attr def {"T: {float, double, int32, int64}"}, // Nodes { {{"two"}, "Const", {}, {{"value", kTwo}, {"dtype", DT_INT64}}}, {{"scale"}, "Cast", {"two"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}}, {{"y"}, "Mul", {"x", "scale"}, {{"T", "$T"}}}, }); } class MlirGraphOptimizationPassTest : public Test { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 08:25:30 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc
ASSERT_NO_FATAL_FAILURE(VerifyConversion<int32>( {1, -1}, DT_INT32, mlir::IntegerType::get(&context, 32))); ASSERT_NO_FATAL_FAILURE(VerifyConversion<int64_t>( {1, -1}, DT_INT64, mlir::IntegerType::get(&context, 64))); ASSERT_NO_FATAL_FAILURE(VerifyConversion<uint4>( {static_cast<uint4>(1), static_cast<uint4>(2)}, DT_UINT4, mlir::IntegerType::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
return DT_UINT16; case toco::IODataType::INT32: return DT_INT32; case toco::IODataType::UINT32: return DT_UINT32; case toco::IODataType::INT64: return DT_INT64; case toco::IODataType::UINT8: return DT_UINT8; case toco::IODataType::UINT64: return DT_UINT64; case toco::IODataType::STRING: return DT_STRING; case toco::IODataType::BOOL:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir
%output = "tf.StridedSlice"(%input,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
ASSERT_EQ(absl::OkStatus(), model_api->GetVariable("sub_module.uninitialized_variable", &uninitialized_variable)); ASSERT_EQ(tensorflow::DT_INT64, uninitialized_variable->dtype()); TF_DeleteSavedModel(saved_model); TF_DeleteStatus(status); TFE_DeleteContext(ctx); } TEST_P(CSavedModelAPITest, LoadSavedModelWithWhileLoop) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0)