- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for _handle_shapes (0.55 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
handle_shapes_attr.mutable_list()->add_shape()); } (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
handle_shapes_attr.mutable_list()->add_shape()); } (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
return errors::InvalidArgument( "Mismatched resource types: ", DataTypeString(a.handle_type), " vs. ", DataTypeString(b.handle_type)); } TF_RETURN_IF_ERROR( a.handle_shape.MergeWith(b.handle_shape, &result.handle_shape)); return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
EXPECT_NE(iter, shape_info.end()); EXPECT_EQ(iter->second.size(), 1); EXPECT_EQ(iter->second.at(0).handle_type, DT_FLOAT); TensorShape resource_shape; EXPECT_TRUE(iter->second.at(0).handle_shape.AsTensorShape(&resource_shape)); EXPECT_EQ(resource_shape, TensorShape({2, 3})); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0)