- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for Bitcast (0.24 sec)
-
tensorflow/c/kernels/ops/bitcast.cc
TF_RegisterOpDefinition(op_builder, status); CHECK_EQ(TF_GetCode(status), TF_OK) << "Bitcast op registration failed: " << TF_Message(status); TF_DeleteStatus(status); } TF_ATTRIBUTE_UNUSED static bool IsBitcastOpRegistered = []() { if ((&TF_NewStatus != nullptr) && SHOULD_REGISTER_OP("Bitcast")) { RegisterBitcastOp(); } return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 07:51:50 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/convert_func_to_bfloat16.mlir
func.func @bitcast_convert_i32_f32(%arg0: tensor<1x256128xi32>) -> tensor<1x256128xf32> { // CHECK: %[[BITCAST:.*]] = stablehlo.bitcast_convert %arg0 : (tensor<1x256128xi32>) -> tensor<1x256128xf32> // CHECK: %[[CONVERT:.*]] = stablehlo.convert %[[BITCAST]] : (tensor<1x256128xf32>) -> tensor<1x256128xbf16> // CHECK: return %[[CONVERT]] : tensor<1x256128xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 08 22:40:14 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
} }; void TestBitcastOp(Tensor* input_tensor, DataType out_type, TensorShape expected_shape, error::Code expected_code) { Status status; NodeDef def; def.set_op("Bitcast"); def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op.cc
{ auto* builder = TF_NewKernelBuilder("Bitcast", tensorflow::DEVICE_CPU, &BitcastOp_Create, &BitcastOp_Compute, &BitcastOp_Delete); TF_RegisterKernelBuilder("BitcastOp", builder, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << "Error while registering bitcast kernel"; } #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 31 03:28:11 UTC 2021 - 5.9K bytes - Viewed (0) -
tensorflow/c/kernels/BUILD
"//tensorflow/c:tf_tensor", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", ], ) tf_gen_op_libs( op_lib_names = ["bitcast"], deps = [ "//tensorflow/c:ops", "//tensorflow/c:tf_datatype", "//tensorflow/c:tf_status", "//tensorflow/c:tf_tensor", "//tensorflow/core:lib", ], )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/op_fuzzing.bzl
], ) op_names = [ "BatchMatrixBandPart", "BatchMatrixDiag", "BatchMatrixDiagPart", "BatchMatrixSetDiag", "BatchToSpace", "BatchToSpaceND", "Bitcast", "BroadcastArgs", "BroadcastTo", "CheckNumerics", "ConcatV2", "ConjugateTranspose", "DebugGradientIdentity", "DeepCopy", "DepthToSpace", "Dequantize",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 07 19:14:57 UTC 2022 - 4.3K bytes - Viewed (0) -
src/cmd/compile/internal/test/testdata/gen/arithBoundaryGen.go
for _, i := range s.i { for _, j := range s.i { fmt.Fprintf(w, "itd%s{a: %d, b: %d, add: %d, sub: %d, mul: %d", s.sn, i, j, icast(i+j, s), icast(i-j, s), icast(i*j, s)) if j != 0 { fmt.Fprintf(w, ", div: %d, mod: %d", icast(i/j, s), icast(i%j, s)) } fmt.Fprint(w, "},\n") } } fmt.Fprintf(w, "}\n") } }
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Sep 20 02:13:02 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
// CHECK-SAME: (tensor<*xf32>) -> tensor<*xf32> // CHECK: %[[round:.*]] = "tf.Round"(%[[avgpool_f32]]) // CHECK: %[[icast:.*]] = "tf.Cast"(%[[round]]) <{Truncate = false}> : (tensor<*xf32>) -> tensor<*xi8> // CHECK: %[[sc2:.*]] = "quantfork.scast"(%[[icast]]) // CHECK: %[[dq:.*]] = "quantfork.dcast"(%[[sc2]]) : (tensor<*x!quant.uniform<i8:f32, 5.000000e-02:-10>>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc
auto dequantize_func = [&](const APInt& ap_int_value) -> APInt { const int64_t int_value = ap_int_value.getSExtValue(); const float real = (int_value - zp) * scale; auto real_int = absl::bit_cast<int32_t>(real); return APInt(/*numBits=*/32, real_int); }; auto dequant_values = mlir::cast<DenseIntOrFPElementsAttr>(input_values)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0)