Searched refs:new_indices (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | tfexample_decoder.py | 335 new_indices = array_ops.concat(
338 tensor = sparse_tensor.SparseTensor(new_indices, values.values, shape)
|
| /external/v8/src/compiler/ |
| D | simd-scalar-lowering.h | 79 void GetIndexNodes(Node* index, Node** new_indices, SimdType type);
|
| D | simd-scalar-lowering.cc | 342 void SimdScalarLowering::GetIndexNodes(Node* index, Node** new_indices, in GetIndexNodes() argument
347 new_indices[laneIndex] = index; in GetIndexNodes()
350 new_indices[laneIndex] = graph()->NewNode( in GetIndexNodes()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_reshape_op_test.py | 305 new_indices = np.transpose(np.nonzero(new_dense < 0.5))
314 self.assertAllEqual(output_val.indices, new_indices)
|
| /external/skia/src/compute/hs/cuda/ |
| D | hs_cuda.inl | 526 hs_indices_t new_indices = 0; local
529 uint32_t const idx = hs_state_acquire(state,&new_indices);
532 indices |= new_indices;
535 uint32_t const wait = before & ~new_indices;
|
| /external/skqp/src/compute/hs/cuda/ |
| D | hs_cuda.inl | 526 hs_indices_t new_indices = 0; local
529 uint32_t const idx = hs_state_acquire(state,&new_indices);
532 indices |= new_indices;
535 uint32_t const wait = before & ~new_indices;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1647 new_indices = array_ops.concat([indices_columns_to_preserve] + ids, 1)
1652 result = sparse_tensor.SparseTensor(new_indices, new_values, new_shape)
1701 new_indices = array_ops.gather(sp_input.indices, where_true)
1703 return sparse_tensor.SparseTensor(new_indices, new_values,
|
| D | array_grad.py | 199 new_indices = array_ops.gather(grad.indices, indices_to_select) - start
201 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
|
| /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)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 470 done_indices, new_indices = data_flow_ops.dynamic_partition(
492 return not_all_done, new_indices, new_inputs, new_output_tas
495 new_indices, cond_stacked, new_inputs, argument
519 body_pfor = self._init_pfor(pfor_input.pfor, new_indices,
620 (not_all_done, new_indices,
624 (not_all_done, new_indices,
632 new_indices, cond_stacked, new_inputs,
639 new_args = ([not_all_done, new_indices] + new_outputs + list(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 493 Array* const new_indices, in ReshapeToRemoveDegenerateDims()
517 new_source, new_indices, new_source_dim, in ReshapeToRemoveDegenerateDims()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 717 DimensionVector new_indices(result_shape.rank()); in SliceInternal() local
721 new_indices[i] = indices[i] + start_indices[i]; in SliceInternal()
723 NativeT value = Get<NativeT>(new_indices); in SliceInternal()
|