Home
last modified time | relevance | path

Searched refs:tensor_util (Results 1 – 25 of 227) sorted by relevance

12345678910

/external/tensorflow/tensorflow/python/framework/
Dtensor_util_test.py30 from tensorflow.python.framework import tensor_util
45 t = tensor_util.make_tensor_proto(value)
51 a = tensor_util.MakeNdarray(t)
56 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0])
69 a = tensor_util.MakeNdarray(t)
74 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0], dtype=dtypes.float32)
87 a = tensor_util.MakeNdarray(t)
92 t = tensor_util.make_tensor_proto([10, 20, 30], dtype=dtypes.float32)
105 a = tensor_util.MakeNdarray(t)
111 t = tensor_util.make_tensor_proto(arr, dtype=dtypes.float32)
[all …]
Dsmart_cond.py18 from tensorflow.python.framework import tensor_util
73 pred_value = tensor_util.constant_value(pred)
77 pred_value = tensor_util.try_evaluate_constant(pred)
/external/executorch/runtime/core/exec_aten/util/test/
Dtargets.bzl15 "//executorch/runtime/core/exec_aten/util:tensor_util",
23 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
25 "//executorch/runtime/core/exec_aten/util:tensor_util",
33 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
34 "//executorch/runtime/core/exec_aten/util:tensor_util",
44 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
46 "//executorch/runtime/core/exec_aten/util:tensor_util" + aten_suffix,
/external/tensorflow/tensorflow/python/autograph/operators/
Dlogical.py17 from tensorflow.python.framework import tensor_util
24 if tensor_util.is_tf_type(a):
42 if tensor_util.is_tf_type(a_val):
61 if tensor_util.is_tf_type(a_val):
79 if tensor_util.is_tf_type(a) or tensor_util.is_tf_type(b):
Dpy_builtins.py34 from tensorflow.python.framework import tensor_util
169 if tensor_util.is_tf_type(x):
194 if tensor_util.is_tf_type(x):
211 if tensor_util.is_tf_type(x):
237 elif tensor_util.is_tf_type(s):
318 if any(tensor_util.is_tf_type(o) for o in objects):
330 if any(tensor_util.is_tf_type(s) for s in args):
362 if any(tensor_util.is_tf_type(s) for s in args):
402 vals = tuple(v.numpy() if tensor_util.is_tf_type(v) else v for v in vals)
414 if any(tensor_util.is_tf_type(s) for s in (start_or_stop, stop, step)):
[all …]
Dslices.py20 from tensorflow.python.framework import tensor_util
55 elif tensor_util.is_tf_type(target):
115 elif tensor_util.is_tf_type(target):
Ddata_structures.py22 from tensorflow.python.framework import tensor_util
105 if tensor_util.is_tf_type(elements):
187 elif tensor_util.is_tf_type(list_):
257 elif tensor_util.is_tf_type(list_):
321 elif tensor_util.is_tf_type(list_):
/external/tensorflow/tensorflow/python/autograph/utils/
Dtensors.py24 from tensorflow.python.framework import tensor_util
30 return (tensor_util.is_tf_type(t) and
43 return (tensor_util.is_tf_type(t) and t.dtype == dtypes.variant and
49 return tensor_util.is_tf_type(t) and hasattr(t, 'op') and t.op.type == 'Range'
Dpy_func.py20 from tensorflow.python.framework import tensor_util
69 arg_is_tensor = tuple(map(tensor_util.is_tf_type, args))
89 kwarg_is_tensor = {k: tensor_util.is_tf_type(kwargs[k]) for k in kwarg_keys}
/external/tensorflow/tensorflow/python/ops/
Dstateless_random_ops.py23 from tensorflow.python.framework import tensor_util
465 shape = tensor_util.shape_tensor(shape)
489 tensor_util.maybe_set_static_shape(result, shape)
552 shape = tensor_util.shape_tensor(shape)
559 tensor_util.maybe_set_static_shape(result, shape)
648 shape = tensor_util.shape_tensor(shape)
659 tensor_util.maybe_set_static_shape(result, shape)
719 shape = tensor_util.shape_tensor(shape)
722 tensor_util.maybe_set_static_shape(result, shape)
762 shape = tensor_util.shape_tensor(shape)
[all …]
Drandom_ops.py23 from tensorflow.python.framework import tensor_util
87 shape_tensor = tensor_util.shape_tensor(shape)
95 tensor_util.maybe_set_static_shape(value, shape)
138 shape_tensor = tensor_util.shape_tensor(shape)
152 tensor_util.maybe_set_static_shape(rnd, shape)
196 shape_tensor = tensor_util.shape_tensor(shape)
204 tensor_util.maybe_set_static_shape(value, shape)
296 shape = tensor_util.shape_tensor(shape)
320 tensor_util.maybe_set_static_shape(result, shape)
554 shape = tensor_util.shape_tensor(shape)
[all …]
Dweights_broadcast_ops.py22 from tensorflow.python.framework import tensor_util
83 weights_rank_static = tensor_util.constant_value(weights_rank)
89 values_rank_static = tensor_util.constant_value(values_rank)
101 weights_shape_static = tensor_util.constant_value(weights_shape)
102 values_shape_static = tensor_util.constant_value(values_shape)
/external/tensorflow/tensorflow/python/ops/signal/
Dreconstruction_ops.py19 from tensorflow.python.framework import tensor_util
60 frame_step_static = tensor_util.constant_value(frame_step)
65 signal_shape_static = tensor_util.constant_value(signal_shape)
73 outer_rank_static = tensor_util.constant_value(outer_rank)
133 perm_static = tensor_util.constant_value(perm)
Dutil_ops.py22 from tensorflow.python.framework import tensor_util
57 const_a = tensor_util.constant_value(a)
58 const_b = tensor_util.constant_value(b)
Dshape_ops.py21 from tensorflow.python.framework import tensor_util
31 frame_length = tensor_util.constant_value(frame_length)
32 frame_step = tensor_util.constant_value(frame_step)
33 axis = tensor_util.constant_value(axis)
130 val_static = tensor_util.constant_value(val)
/external/tensorflow/tensorflow/python/ops/distributions/
Dtransformed_distribution.py21 from tensorflow.python.framework import tensor_util
43 return tensor_util.constant_value(ops.convert_to_tensor(x))
281 tensor_util.constant_value(self._override_batch_shape) is None or
282 tensor_util.constant_value(self._override_batch_shape).size != 0)
290 tensor_util.constant_value(self._override_event_shape) is None or
291 tensor_util.constant_value(self._override_event_shape).size != 0)
351 static_override = tensor_util.constant_value_as_shape(
374 static_override = tensor_util.constant_value_as_shape(
584 if tensor_util.constant_value(override_is_scalar):
597 if tensor_util.constant_value(override_shape) is not None:
[all …]
/external/executorch/kernels/quantized/test/
Dtargets.bzl16 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
26 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
34 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
40 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
/external/tensorflow/tensorflow/python/kernel_tests/summary_ops/
Dsummary_v1_tensor_op_test.py22 from tensorflow.python.framework import tensor_util
69 n = tensor_util.MakeNdarray(value.tensor)
80 n = tensor_util.MakeNdarray(value.tensor)
89 n = tensor_util.MakeNdarray(value.tensor)
99 n = tensor_util.MakeNdarray(value.tensor)
110 n = tensor_util.MakeNdarray(value.tensor)
/external/tensorflow/tensorflow/python/ops/linalg/
Dslicing.py22 from tensorflow.python.framework import tensor_util
32 constant_args = [tensor_util.constant_value(a) for a in args]
138 if (tensor_util.constant_value(array_ops.size(batch_shape)) != 0 and
139 tensor_util.constant_value(array_ops.size(param_batch_shape)) == 0):
149 slc for slc in slices if not tensor_util.is_tensor(slc)]:
/external/executorch/runtime/core/exec_aten/testing_util/
Dtargets.bzl13 name = "tensor_util" + (aten_suffix),
14 srcs = ["tensor_util.cpp"],
16 "tensor_util.h",
45 "//executorch/runtime/core/exec_aten/util:tensor_util" + aten_suffix,
/external/tensorflow/tensorflow/python/autograph/lang/
Dspecial_functions.py24 from tensorflow.python.framework import tensor_util
31 if tensor_util.is_tf_type(elements):
48 if tensor_util.is_tf_type(like_value):
/external/tensorflow/tensorflow/core/runtime_fallback/util/
DBUILD38 "//tensorflow/core/tfrt/utils:tensor_util",
94 name = "tensor_util",
95 srcs = ["tensor_util.cc"],
97 "tensor_util.h",
138 ":tensor_util",
180 ":tensor_util",
/external/tensorflow/tensorflow/tools/graph_transforms/python/
Dtransform_graph_test.py20 from tensorflow.python.framework import tensor_util
39 attr_value_pb2.AttrValue(tensor=tensor_util.make_tensor_proto(
48 attr_value_pb2.AttrValue(tensor=tensor_util.make_tensor_proto(
/external/tensorflow/tensorflow/python/distribute/
Dsummary_op_util.py20 from tensorflow.python.framework import tensor_util
43 replica_id = tensor_util.constant_value(replica_id)
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
DBUILD92 name = "tensor_util",
93 srcs = ["tensor_util.cc"],
94 hdrs = ["tensor_util.h"],
182 "//tensorflow/core/runtime_fallback/util:tensor_util",
216 ":tensor_util",
218 "//tensorflow/core/runtime_fallback/util:tensor_util",
264 ":tensor_util",
269 "//tensorflow/core/runtime_fallback/util:tensor_util",
366 "//tensorflow/core/tfrt/utils:tensor_util",

12345678910