- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 96 for bitcast (0.41 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/c/kernels/bitcast_op.cc
// // Bitcast is implemented as a low-level cast, so machines with different endian // orderings will give different results. typedef struct BitcastOp { TF_DataType input_data_type; TF_DataType output_data_type; size_t in_size; size_t out_size; } BitcastOp; static void* BitcastOp_Create(TF_OpKernelConstruction* ctx) { auto* kernel = new BitcastOp; TF_Status* s = TF_NewStatus();
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/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/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) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: return %[[RES0]] : tensor<8xf32> } func.func @bitcast(%arg0: tensor<8xi32>) -> tensor<8xui32> { %0 = "tf.Bitcast"(%arg0) : (tensor<8xi32>) -> tensor<8xui32> func.return %0 : tensor<8xui32> // CHECK-LABEL: bitcast // CHECK: %[[RES0:.*]] = "tfl.bitcast"(%arg0) : (tensor<8xi32>) -> tensor<8xui32> // CHECK: return %[[RES0]] : tensor<8xui32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
func.func @testDoubleBitcast(%arg0: tensor<8x16x32x64xf32>) -> tensor<8x16x32x64xi32> { %0 = "tf.Bitcast"(%arg0) : (tensor<8x16x32x64xf32>) -> tensor<8x16x32x64x2xi16> %1 = "tf.Bitcast"(%0) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<8x16x32x64x2xi16>) -> tensor<8x16x32x64xi32> func.return %1: tensor<8x16x32x64xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize.td
(TF_BiasAddOp:$dest $arg0, $arg1, ConstantAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">), [], [(CopyAttrs $src, $dest)]>; //===----------------------------------------------------------------------===// // Bitcast op patterns. //===----------------------------------------------------------------------===// // Parameter attributes are part of function attributes, which is not affected
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:42:28 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.func @testBitcast(%arg0: tensor<3x4xui16>) -> tensor<3x4x!tf_type.quint16> { %0 = "tf.Bitcast"(%arg0) : (tensor<3x4xui16>) -> tensor<3x4x!tf_type.quint16> func.return %0 : tensor<3x4x!tf_type.quint16> } // ----- // CHECK-LABEL: func @testBitcast_v2 func.func @testBitcast_v2(%arg0: tensor<3x2xi16>) -> tensor<3xi32> { %0 = "tf.Bitcast"(%arg0) : (tensor<3x2xi16>) -> tensor<3xi32> func.return %0 : tensor<3xi32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)