Searched refs:v_batch (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | diag_op_test.py | 397 v_batch = np.array([[1.0, 0.0, 3.0], [4.0, 5.0, 6.0]]).astype(dtype)
401 v_batch_diag = array_ops.matrix_diag(v_batch)
407 v_batch_diag = array_ops.matrix_diag(v_batch, k=offset)
409 np.diag(v_batch[i], offset) for i in range(0, v_batch.shape[0])
631 v_batch = np.array([[-1.0, 0.0, -3.0], [-4.0, -5.0, -6.0]]).astype(dtype)
641 output = array_ops.matrix_set_diag(mat_batch, v_batch)
668 v_batch = np.array([[-1.0, -2.0], [-4.0, -5.0]])
674 output = array_ops.matrix_set_diag(mat_batch, v_batch)
817 v_batch = np.array([[1.0, 0.0, 3.0], [4.0, 5.0, 6.0]]).astype(dtype)
824 self.assertAllEqual(mat_batch_diag, v_batch)
[all …]
|
| /external/mesa3d/src/gallium/drivers/iris/ |
| D | iris_batch.c | 123 decode_get_bo(void *v_batch, bool ppgtt, uint64_t address) in decode_get_bo() argument
125 struct iris_batch *batch = v_batch; in decode_get_bo()
148 decode_get_state_size(void *v_batch, in decode_get_state_size() argument
152 struct iris_batch *batch = v_batch; in decode_get_state_size()
|
| /external/mesa3d/src/intel/vulkan/ |
| D | anv_device.c | 2606 decode_get_bo(void *v_batch, bool ppgtt, uint64_t address) in decode_get_bo() argument
2608 struct anv_device *device = v_batch; in decode_get_bo()
|