Home
last modified time | relevance | path

Searched refs:type_inference (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/
Dtype_inference_test.py26 from tensorflow.python.autograph.pyct.static_analysis import type_inference
30 class BasicTestResolver(type_inference.Resolver):
63 node = type_inference.resolve(node, ctx, graphs, self.resolver)
84 class Resolver(type_inference.Resolver):
103 class Resolver(type_inference.Resolver):
119 class Resolver(type_inference.Resolver):
140 class Resolver(type_inference.Resolver):
193 class Resolver(type_inference.Resolver):
245 class Resolver(type_inference.Resolver):
283 class Resolver(type_inference.Resolver):
[all …]
DBUILD27 "type_inference.py",
/external/pytorch/torch/csrc/utils/
Dtensor_new.cpp297 bool type_inference, in internal_new_from_data() argument
316 type_inference ? var.scalar_type() : scalar_type; in internal_new_from_data()
334 type_inference ? tensor.scalar_type() : scalar_type; in internal_new_from_data()
360 type_inference ? tensor.scalar_type() : scalar_type; in internal_new_from_data()
377 type_inference ? infer_scalar_type(data) : scalar_type; in internal_new_from_data()
950 bool type_inference = r.isNone(ARG_TYPE); in sparse_compressed_tensor_ctor_worker() local
967 /*type_inference=*/type_inference); in sparse_compressed_tensor_ctor_worker()
1005 bool type_inference = r.isNone(ARG_TYPE1); in sparse_compressed_tensor_ctor_worker() local
1023 /*type_inference=*/type_inference); in sparse_compressed_tensor_ctor_worker()
1177 bool type_inference = r.isNone(ARG_TYPE); in sparse_coo_tensor_ctor() local
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op_test.cc218 static Status type_inference(Graph& graph) { in type_inference() function
276 TF_EXPECT_OK(type_inference(graph)); in TEST_F()
326 TF_EXPECT_OK(type_inference(graph)); in TEST_F()
384 TF_EXPECT_OK(type_inference(graph)); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops_test.cc149 static Status type_inference(Graph& graph) { in type_inference() function
203 TF_EXPECT_OK(type_inference(graph)); in TEST()
/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op_test.cc319 static Status type_inference(Graph& graph) { in type_inference() function
377 TF_EXPECT_OK(type_inference(graph)); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen.py40 from tensorflow.python.autograph.pyct.static_analysis import type_inference
359 class TFRTypeResolver(type_inference.Resolver):
1508 node = type_inference.resolve(node, ctx, graphs,
/external/pytorch/torch/_refs/
D__init__.py6455 type_inference, argument
6465 inferred_scalar_type = var.dtype if type_inference else scalar_type
6481 inferred_scalar_type = _infer_scalar_type(data) if type_inference else scalar_type
6512 type_inference = dtype is None
6522 type_inference=type_inference,