Searched refs:normalized_values (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/util/protobuf/ |
| D | compare.py | 142 normalized_values = None
150 normalized_values = [int(x) for x in values]
155 normalized_values = [int(x) for x in values]
157 normalized_values = [round(x, 6) for x in values]
159 normalized_values = [round(float(x), 7) for x in values]
161 if normalized_values is not None:
164 getattr(pb, desc.name).extend(normalized_values)
166 setattr(pb, desc.name, normalized_values[0])
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | SpmmReduceKernel.cpp | 349 const Tensor& normalized_values, in spmm_reduce_normalize_values_kernel_impl() argument
359 auto normalized_values_data = normalized_values.accessor<scalar_t, 1>(); in spmm_reduce_normalize_values_kernel_impl()
501 const Tensor& normalized_values, in spmm_reduce_normalize_values_kernel() argument
508 normalized_values, values, crow_indices, row_indices); in spmm_reduce_normalize_values_kernel()
530 Tensor normalized_values = at::empty(values.sizes(), values.options()); in spmm_reduce_backward_other_kernel() local
531 spmm_reduce_normalize_values_kernel(normalized_values, values, crow_indices, row_indices); in spmm_reduce_backward_other_kernel()
532 val = normalized_values.index_select(0, csr2csc); in spmm_reduce_backward_other_kernel()
|