Searched refs:weights0 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1921 auto weights0 = builder.AddInstruction( in TEST_F() local
1939 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
1962 EXPECT_EQ(assignment->GetUniqueSlice(weights0, {}).ConsumeValueOrDie(), in TEST_F()
2244 auto weights0 = builder.AddInstruction( in TEST_F() local
2258 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
2386 auto weights0 = builder.AddInstruction( in TEST_F() local
2403 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
2438 input0, weights0, zero, output0, while0->mutable_operand(0), while0, in TEST_F()
2462 auto weights0 = builder.AddInstruction( in TEST_F() local
2478 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms.cc | 249 Tensor weights0 = GetNodeTensorAttr(weights0_node, "value"); in FuseBatchNormWithConvConcat() local
250 const int64 split_cols = weights0.shape().dim_size(3); in FuseBatchNormWithConvConcat()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1390 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
1398 vmull_s8(vget_low_s8(weights0), vget_low_s8(input)); in ShuffledFullyConnectedWorkerImpl()
1406 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input)); in ShuffledFullyConnectedWorkerImpl()
1471 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
1483 local_accum0 = vmull_s8(vget_low_s8(weights0), vget_low_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
1488 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1334 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
1342 vmull_s8(vget_low_s8(weights0), vget_low_s8(input)); in ShuffledFullyConnectedWorkerImpl()
1350 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input)); in ShuffledFullyConnectedWorkerImpl()
1415 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
1427 local_accum0 = vmull_s8(vget_low_s8(weights0), vget_low_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
1432 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
|