Home
last modified time | relevance | path

Searched refs:tensor_values (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback_test.py154 tensor_values = collections.defaultdict(lambda: [])
212 tensor_values[execution.op_type].append(
220 self.assertAllClose(tensor_values["Greater"], [1, 1, 1, 1, 1, 1, 0])
221 self.assertAllClose(tensor_values["RealDiv"], [5, 8, 4, 2, 1])
222 self.assertAllClose(tensor_values["Mul"], [15])
223 self.assertAllClose(tensor_values["AddV2"], [16])
924 tensor_values = [reader.graph_execution_trace_to_tensor_value(digest)
928 self.assertLen(tensor_values, 2)
929 self.assertAllClose(tensor_values[0], 5.0) # 1st AddV2 op.
931 tensor_values[1], np.log(5.0) + 1.0) # 2nd AddV2 op.
[all …]
H A Ddebug_events_monitors.py265 tensor_values = self._debug_data_reader.execution_to_tensor_values(
267 for output_slot, tensor_value in enumerate(tensor_values):
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_supported_values_test.py127 tensor_values = constant_op.constant(
129 values = WrappedTensor(tensor_values)
134 self.assertAllEqual(rt.values.value, tensor_values)
140 self.assertAllEqual(rt.values.value, tensor_values)
146 self.assertAllEqual(rt.values.value, tensor_values)
152 self.assertAllEqual(rt.values.value, tensor_values)
157 self.assertAllEqual(rt.values.value, tensor_values)
161 tensor_values = constant_op.constant(['a', 'b', 'c', 'd', 'e', 'f', 'g'])
162 values = WrappedTensor(tensor_values)
175 tensor_values = constant_op.constant(['a', 'b', 'c', 'd', 'e', 'f', 'g'])
[all …]
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dappend_slices_test.py49 tensor_values = [tf.constant(x, dtype=tf.int32) for x in (7, 11)]
50 for (tensor_name, tensor_value) in zip(tensor_names, tensor_values):
146 tensor_values = [tf.constant(x, dtype=tf.int32) for x in (7, 11)]
147 for tensor_value in tensor_values:
165 tensor_values = [tf.constant(x, dtype=tf.int32) for x in (7, 11)]
166 for (tensor_name, tensor_value) in zip(tensor_names, tensor_values):
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dop_kernel_runner.h106 const gtl::InlinedVector<tensorflow::TensorValue, 4>& tensor_values, in OpKernelRunState()
112 input_tf_tensors.reserve(tensor_values.size()); in OpKernelRunState()
113 for (const auto& tensor_value : tensor_values) { in OpKernelRunState()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dcalibration_logger.h53 const float* tensor_values, size_t tensor_size, in LogTensorValue() argument
56 return tensor_id_to_stats_map_[key].Update(tensor_values, tensor_size, in LogTensorValue()
/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py411 tensor_values: Dict[str, TensorMeta] = field(default_factory=dict) variable in GraphState
450 self.graph_state.tensor_values[node.name] = serialize_tensor_meta(
901 assert name not in self.graph_state.tensor_values
902 self.graph_state.tensor_values[name] = serialize_tensor_meta(meta_val)
1328 tensor_values=self.graph_state.tensor_values,
1584 for name, tensor_value in serialized_graph.tensor_values.items():
2644 a.name = _rename(a.name, graph.tensor_values)
2691 sorted(graph.tensor_values.items(), key=operator.itemgetter(0))
2716 tensor_values=sorted_tensor_values,
H A Dschema.py209 tensor_values: Dict[str, TensorMeta]
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession.py532 def build_results(self, session, tensor_values): argument
552 assert len(self._final_fetches) == len(tensor_values)
568 value = tensor_values[j]
576 assert j == len(tensor_values)
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dbasic_session_run_hooks.py252 def _log_tensors(self, tensor_values): argument
257 logging.info(self._formatter(tensor_values))
261 stats.append("%s = %s" % (tag, tensor_values[tag]))
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py407 tensor_values: Dict[str, TensorMeta] = field(default_factory=dict) variable in GraphState
448 self.graph_state.tensor_values[node.name] = serialize_tensor_meta(
931 assert name not in self.graph_state.tensor_values
932 self.graph_state.tensor_values[name] = serialize_tensor_meta(meta_val)
1323 tensor_values=self.graph_state.tensor_values,
1611 for name, tensor_value in serialized_graph.tensor_values.items():
2636 a.name = _rename(a.name, graph.tensor_values)
2682 sorted_tensor_values = dict(sorted(graph.tensor_values.items(), key=operator.itemgetter(0)))
2706 tensor_values=sorted_tensor_values,
H A Dschema.py201 tensor_values: Dict[str, TensorMeta]
/aosp_15_r20/external/pytorch/test/
H A Dtest_foreach.py376 tensor_values = torch.tensor(scalars)
388 scalars=tensor_values,
397 scalars=tensor_values[0],
408 scalars=tensor_values.cuda(),
418 scalars=tensor_values[:2],
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc88 typename checkpoint::SaveTypeTraits<T>::RepeatedField* tensor_values = in ValuesFromConstNode() local
91 if (!tensor_values->empty() && tensor.has_tensor_shape()) { in ValuesFromConstNode()
96 if (shape.dim_size() == 1 && shape.dim(0).size() == tensor_values->size()) { in ValuesFromConstNode()
97 values->insert(values->end(), tensor_values->begin(), in ValuesFromConstNode()
98 tensor_values->end()); in ValuesFromConstNode()