Searched refs:indices2 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_nd_op_test.py | 212 indices2 = array_ops.placeholder(dtypes.int32, shape=[None])
219 ragged_gather_ops.gather_nd(params, indices2)
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 106 Eigen::DSizes<ptrdiff_t, 5> indices2(1,1,3,4,5); in test_simple_slice() local
108 slice2 = tensor.slice(indices2, sizes2); in test_simple_slice()
147 Eigen::DSizes<ptrdiff_t, 2> indices2(0,2); in test_slice_in_expr() local
149 …tensor3 = tensor1.slice(indices1, sizes1).contract(tensor2.slice(indices2, sizes2), contract_along… in test_slice_in_expr()
|
| D | cxx11_tensor_ref.cpp | 101 Eigen::DSizes<ptrdiff_t, 5> indices2(1,1,3,4,5); in test_slice() local
103 slice = tensor.slice(indices2, sizes2); in test_slice()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_convert_test.cc | 162 input: "indices2" in TEST()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2_test.py | 91 indices2 = constant_op.constant(index_values2, name="i2")
92 return array_ops.gather(y, indices2, name="y2")
|
| D | gradient_checker_test.py | 131 indices2 = constant_op.constant(index_values2, name="i2")
132 y2 = array_ops.gather(y, indices2, name="y2")
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/jitter/ |
| D | fetch_jit.cpp | 189 Value* indices2; in Create() local
197 indices2 = GEP(indices, C(8)); in Create()
204 …vIndices2 = LOAD(BITCAST(indices2, PointerType::get(VectorType::get(mInt8Ty, mpJitMgr->mVWidth), 0… in Create()
214 vIndices2 = GetSimdValid8bitIndices(indices2, pLastIndex); in Create()
221 indices2 = GEP(indices, C(8)); in Create()
228 …vIndices2 = LOAD(BITCAST(indices2, PointerType::get(VectorType::get(mInt16Ty, mpJitMgr->mVWidth), … in Create()
238 vIndices2 = GetSimdValid16bitIndices(indices2, pLastIndex); in Create()
244 indices2 = GEP(indices, C(8)); in Create()
249 …(fetchState.bDisableIndexOOBCheck) ? vIndices2 = LOAD(BITCAST(indices2, PointerType::get(mSimdInt3… in Create()
250 … : vIndices2 = GetSimdValid32bitIndices(indices2, pLastIndex); in Create()
|