- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 61 for Dtype (0.04 sec)
-
tensorflow/compiler/jit/xla_device.cc
Status DefaultPaddedShapeFn(const Tensor& tensor, xla::Shape* shape) { const tensorflow::XlaTensor* xla_tensor = tensorflow::XlaTensor::FromTensor(&tensor); if (xla_tensor == nullptr) { return TensorShapeToXLAShape(tensor.dtype(), tensor.shape(), shape); } const xla::ShapedBuffer& shaped_buffer = xla_tensor->shaped_buffer(); *shape = shaped_buffer.on_device_shape(); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K 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/xla_launch_util_test.cc
CreateContext(); std::vector<XlaCompiler::Argument> args(2); args[0].kind = XlaCompiler::Argument::kParameter; args[0].type = DT_INT32; args[0].shape = TensorShape({1, 3}); args[1].kind = XlaCompiler::Argument::kParameter; args[1].type = DT_INT32; args[1].shape = TensorShape({1, 3}); const XlaCompiler::CompilationResult* result; xla::PjRtLoadedExecutable* executable;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
handle_methods.deallocator = &ParallelTensorDeallocator; handle_methods.summarize = &ParallelTensorSummarize; return TensorHandlePtr(TFE_NewCustomDeviceTensorHandle( context, parallel_device_name.c_str(), t_released->dtype(), t_released, handle_methods, status)); } // For TFE_CustomDevice::copy_tensor_to_device in the parallel device // registration. // // Since this function is used to satisfy the TFE_CustomDevice C API,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 29 22:05:31 UTC 2023 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
<< compilation_result.xla_output_shape.ToString() << '\n'; for (const auto& xla_output_description : compilation_result.outputs) { output << "// XlaOutputDescription type=" << DataTypeString(xla_output_description.type) << " shape=(" << absl::StrJoin(xla_output_description.shape.dim_sizes(), ", ") << ')'; if (xla_output_description.input_index >= 0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K 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/c/eager/gradients.cc
gtl::ArraySlice<AbstractTensorHandle*> output_gradients, absl::Span<AbstractTensorHandle*> result) const override; // Builds a tensor filled with ones with the same shape and dtype as `t`. Status BuildOnesLike(const TapeTensor& t, AbstractTensorHandle** result) const override; // Looks up the ID of a Gradient.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
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/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)