- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 169 for Dtype (0.04 sec)
-
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
// Args {"x: T"}, // Return values {"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 {
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/cc/experimental/libtf/tests/function_test.cc
StatusOr<TaggedValue> v = tf_function.Execute(ctx_.get(), args); ASSERT_TRUE(tensorflow::errors::IsInternal(v.status())) << v.status(); ASSERT_TRUE( absl::StrContains(v.status().message(), "Shape and dtype of tensor")); ASSERT_TRUE(absl::StrContains(v.status().message(), "does not match that in signature")); } INSTANTIATE_TEST_SUITE_P(TF2CAPI, FunctionTest,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 21:44:52 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
}; TF_EXPECT_OK(ShapeAnnotationsMatch(graph, shape_info, expected)); } TEST(ShapeInferenceTest, WhileLoopWithResource) { // Graph: // x = resource_variable_ops.var_handle_op(dtype=dtypes.float32, shape=[2, 3]) // y = control_flow_ops.while_loop(lambda _: true, lambda x: x, [x]) Graph graph(OpRegistry::Global()); { Scope scope = Scope::NewRootScope().ExitOnError(); auto x =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema.fbs
// When set, quantized_bias_type defines the dtype for both bias and accumulator. quantized_bias_type: TensorType; } table ExpandDimsOptions { } table SparseToDenseOptions { validate_indices:bool; } table EqualOptions { } table NotEqualOptions { } table ShapeOptions { // Optional output type of the operation (int32 or int64). Defaults to int32.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/tf_saved_model_api.cc
kSavedModelVariablesFilename); ImmediateTensorHandlePtr restored_output; TF_RETURN_IF_ERROR(internal::SingleRestore( context, variables_path_prefix, checkpoint_key, variable->dtype(), &restored_output)); // Assign the restored tensor's value to the variable return variable->Assign(restored_output.get()); })); return Status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
auto iter = placeholders.find(placeholder_index); Node* placeholder_node; if (iter == placeholders.end()) { NodeDefBuilder placeholder_builder(new_name, "Placeholder"); placeholder_builder.Attr("dtype", src->output_type(src_output)); string outside_compilation_attr; TF_RETURN_IF_ERROR(GetNodeAttr(dst->attrs(), outside_compilation_attr_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-dynamic-layout-pass.mlir
} // ----- // Tests that the pass can transform execution with model parallelism and no // replication. // // The following TPUCompileMetadataProto is used: // args { // dtype: DT_FLOAT // shape { // dim { // size: 128 // } // } // } // num_replicas: 1 // num_cores_per_replica: 2 // CHECK-LABEL: func @parallel_execute
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
// First create a simple identity function. TF_Graph* function_graph = TF_NewGraph(); TF_OperationDescription* arg_descr = TF_NewOperation(function_graph, "Placeholder", "arg"); TF_SetAttrType(arg_descr, "dtype", TF_INT32); TF_Status* status = TF_NewStatus(); TF_Operation* arg = TF_FinishOperation(arg_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/c/kernels.h
TF_Status* status); // Allocates a temporary Tensor of the specified type and shape. The // Tensor must not be used after kernel construction is // complete. // // num_dims must equal the size of array dims TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTemp( TF_OpKernelContext* context, TF_DataType dtype, const int64_t* dims, int num_dims, TF_AllocatorAttributes* alloc_attrs, TF_Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
// Specify this trait if the op requires all outputs to have the same type and // the inputs either have the same type as result or a ref type corresponding to // the result type. def TF_OperandsSameAsResultsTypeOrRef : NativeOpTrait< "TF::OperandsSameAsResultsTypeOrRef">; // Op has the same operand and result element types (or type itself, if scalar) // after resolving reference types (i.e., after converting reference types to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0)