Searched refs:x_indices (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_bicubic_op_test.cc | 138 std::array<int64, 4> x_indices; in ResizeBicubicBaseline() local
140 &x_indices); in ResizeBicubicBaseline()
146 {static_cast<float>(images(b, y_indices[i], x_indices[0], c)), in ResizeBicubicBaseline()
147 static_cast<float>(images(b, y_indices[i], x_indices[1], c)), in ResizeBicubicBaseline()
148 static_cast<float>(images(b, y_indices[i], x_indices[2], c)), in ResizeBicubicBaseline()
150 images(b, y_indices[i], x_indices[3], c))}}; in ResizeBicubicBaseline()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_test.py | 65 x_indices = np.vstack(np.where(x)).astype(indices_dtype).T
71 indices=x_indices, values=x_values, dense_shape=x_shape)
114 x_indices = np.vstack(np.where(x)).astype(np.int64).T
119 x_st = sparse_tensor.SparseTensor(x_indices, x_values, x_shape)
124 x_st_shape_unknown = sparse_tensor.SparseTensor(x_indices, x_values,
133 x_indices, x_values, x_shape_inconsistent)
|
| D | sparse_tensor_dense_matmul_grad_test.py | 38 x_indices = np.vstack(non_zero).astype(indices_dtype).T
43 indices=x_indices, values=x_values, dense_shape=x_shape), len(x_values)
|
| D | sparse_conditional_accumulator_test.py | 490 x_indices = array_ops.placeholder(dtypes_lib.int64)
493 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
500 x_indices: [[0, 1], [1, 0]],
510 x_indices = array_ops.placeholder(dtypes_lib.int64)
513 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
519 x_indices: [0, 1],
|
| D | sparse_add_op_test.py | 43 x_indices = np.vstack(non_zero).astype(index_dtype).T
48 indices=x_indices, values=x_values, dense_shape=x_shape), len(x_values)
|
| D | shape_ops_test.py | 43 x_indices = np.vstack(non_zero).astype(index_dtype).T
48 indices=x_indices, values=x_values, dense_shape=x_shape), len(x_values)
|
| D | cwise_ops_unary_test.py | 50 x_indices = np.vstack(non_zero).astype(index_dtype).T
55 indices=x_indices, values=x_values, dense_shape=x_shape), x_values
|
| D | sparse_ops_test.py | 45 x_indices = np.vstack(non_zero).astype(index_dtype).T
50 indices=x_indices, values=x_values, dense_shape=x_shape), len(x_values)
|
| D | cwise_ops_binary_test.py | 53 x_indices = np.vstack(non_zero).astype(index_dtype).T
58 indices=x_indices, values=x_values, dense_shape=x_shape), x_values
|
| D | cwise_ops_test.py | 60 x_indices = np.vstack(non_zero).astype(index_dtype).T
65 indices=x_indices, values=x_values, dense_shape=x_shape), x_values
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 212 x_indices = op.inputs[0]
223 scaled_indices = x_indices // scaling
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 146 jintArray x_indices, jlongArray dx_handles, jintArray dx_indices) { in Java_org_tensorflow_Graph_addGradients() argument
159 resolveOutputs(env, "x", x_handles, x_indices, x.get(), nx); in Java_org_tensorflow_Graph_addGradients()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1462 Value x_indices = rewriter.create<ReshapeOp>( in matchAndRewrite() local
1475 y_indices, x_indices); in matchAndRewrite()
|