/external/tensorflow/tensorflow/core/kernels/ |
sparse_concat_op.cc | 42 OpInputList inds; variable 43 OP_REQUIRES_OK(context, context->input_list("indices", &inds)); 44 const int N = inds.size(); 46 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(inds[i].shape()), 49 inds[i].shape().DebugString(), " at position ", i)); 127 sp_inputs.emplace_back(tensor::DeepCopy(inds[i]),
|
/external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
masked_matmul_ops_test.py | 34 inds = [[0, 0], [0, 2], [1, 1], [2, 0], [2, 3]] * 100 35 indices = np.array(inds).astype(np.int64)
|
/external/swiftshader/third_party/LLVM/lib/VMCore/ |
ConstantsContext.h | 308 ArrayRef<unsigned> inds = ArrayRef<unsigned>()) 310 operands(ops.begin(), ops.end()), indices(inds.begin(), inds.end()) {}
|
/external/mesa3d/src/gallium/auxiliary/draw/ |
draw_llvm.c | 878 LLVMValueRef inds[LP_MAX_VECTOR_WIDTH / 32]; local 889 inds[i] = indices[i]; 891 inds[i] = linear_inds[i]; 893 io_ptrs[i] = LLVMBuildGEP(builder, io_ptr, &inds[i], 1, ""); 1023 LLVMValueRef inds[LP_MAX_VECTOR_WIDTH \/ 32]; local [all...] |
/external/tensorflow/tensorflow/python/ops/ |
sparse_ops.py | 220 inds = [sp_input.indices for sp_input in sp_inputs] 237 gen_sparse_ops._sparse_concat(inds, vals, shapes, axis, name=name)) [all...] |
/external/tensorflow/tensorflow/python/kernel_tests/ |
metrics_test.py | [all...] |
/external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
metric_ops_test.py | [all...] |