Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:updates

33     updates = [100., 200.]
37 tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run()
46 updates = [100., 200.]
50 tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run()
58 updates = []
62 tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run()
69 updates = [100.]
73 'Number of updates should be same as number of indices.'):
74 tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run()
81 updates = [[100., 200., 300.], [400., 500., 600.]]
85 tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run()