Searched refs:new_indices (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1208 new_indices = array_ops.concat([indices_columns_to_preserve] + ids, 1)
1213 result = sparse_tensor.SparseTensor(new_indices, new_values, new_shape)
1255 new_indices = array_ops.gather(sp_input.indices, where_true)
1257 return sparse_tensor.SparseTensor(new_indices, new_values,
|
| D | array_grad.py | 202 new_indices = array_ops.gather(grad.indices, indices_to_select) - start
204 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 822 DimensionVector new_indices(ShapeUtil::Rank(result_shape)); in Slice() local
828 new_indices[i] = indices[i] + start_indices[i]; in Slice()
830 float value = Get<float>(new_indices); in Slice()
838 new_indices[i] = indices[i] + start_indices[i]; in Slice()
840 complex64 value = Get<complex64>(new_indices); in Slice()
848 new_indices[i] = indices[i] + start_indices[i]; in Slice()
850 int32 value = Get<int32>(new_indices); in Slice()
858 new_indices[i] = indices[i] + start_indices[i]; in Slice()
860 uint32 value = Get<uint32>(new_indices); in Slice()
|
| /external/v8/src/compiler/ |
| D | simd-scalar-lowering.cc | 140 void SimdScalarLowering::GetIndexNodes(Node* index, Node** new_indices) { in GetIndexNodes() argument
141 new_indices[0] = index; in GetIndexNodes()
143 new_indices[i] = graph()->NewNode(machine()->Int32Add(), index, in GetIndexNodes()
|
| D | simd-scalar-lowering.h | 57 void GetIndexNodes(Node* index, Node** new_indices);
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | tfexample_decoder.py | 334 new_indices = array_ops.concat(
337 tensor = sparse_tensor.SparseTensor(new_indices, values.values, shape)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_reshape_op_test.py | 287 new_indices = np.transpose(np.nonzero(new_dense < 0.5))
296 self.assertAllEqual(output_val.indices, new_indices)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals_test.py | 77 new_indices = array_ops.concat([new_row_ids, old_col_ids], 1)
79 indices=new_indices, values=sp_x.values, dense_shape=shape)
|