| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | lstm_ops.cc | 328 const Tensor* b_tensor = nullptr; in Compute() local
545 const Tensor* b_tensor = nullptr; in Compute() local
986 const Tensor* b_tensor = nullptr; in Compute() local
1189 const Tensor* b_tensor = nullptr; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_nodes_test.cc | 217 Tensor b_tensor(DT_FLOAT, TensorShape({k, n})); in TestQuantizeMatMul() local
380 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2, 3})); in TestQuantizeConcat() local
1239 Tensor b_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicateConsts() local
1286 Tensor b_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesNested() local
1336 Tensor b_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesInOut() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | min_max_builder.cc | 29 const auto& b_tensor = context->tensors[b_tensor_id]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_list_util.cc | 59 const Tensor& b_tensor = b.tensors()[i]; in TensorListBinaryAdd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | optional_ops_util.cc | 70 const Tensor& b_tensor = b.get_values()[i]; in OptionalBinaryAdd() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | BatchNormalizationLayerFusionFixture.h | 66 …void fill(U &&src, U &&w_tensor, U &&b_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U… in fill()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | message_wrappers_test.cc | 35 Tensor b_tensor(DT_INT32, TensorShape({1, 2})); in TensorB() local
|
| H A D | master_test.cc | 307 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | grpc_session_debug_test.cc | 60 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1})); in CreateGraphDef() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | atest.cpp | 118 auto b_tensor = tensor({b}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session_test.cc | 52 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1})); in CreateGraphDef() local
420 Tensor b_tensor(DT_FLOAT, TensorShape({kSize, 1})); in TEST() local
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_where_test.cpp | 57 …const auto b_tensor = tf_b.make(sizes, /*data=*/{ 6, 5, 4, 3, 2, 1, 1, 2, 3, 4, 5, 6}); in test_where() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/ |
| H A D | impl.cpp | 223 const auto b_tensor = is_broadcast_input_1 ? src1 : src0; in add_sub_q8_neon_fixedpoint() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuMulKernel.cpp | 377 const auto b_tensor = is_broadcast_input_1 ? src1 : src0; in mul_q8_neon_fixedpoint() local
|