Searched refs:weights0 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms.cc | 248 Tensor weights0 = GetNodeTensorAttr(weights0_node, "value"); in FuseBatchNormWithConvConcat() local
249 const int64 split_cols = weights0.shape().dim_size(3); in FuseBatchNormWithConvConcat()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 2069 auto weights0 = builder.AddInstruction( in TEST_F() local
2087 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
2110 EXPECT_EQ(assignment->GetUniqueSlice(weights0, {}).ConsumeValueOrDie(), in TEST_F()
2391 auto weights0 = builder.AddInstruction( in TEST_F() local
2405 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
2568 auto weights0 = builder.AddInstruction( in TEST_F() local
2585 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
2620 input0, weights0, zero, output0, while0->mutable_operand(0), while0, in TEST_F()
2643 auto weights0 = builder.AddInstruction( in TEST_F() local
2659 HloInstruction::CreateTuple({input0, weights0, output0})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_test.py | 851 weights0 = self._get_replica_numpy(weights_activations, strategy, 0)
862 self.assertAllClose(golden, weights0)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 514 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
522 vmull_s8(vget_low_s8(weights0), vget_low_s8(input)); in ShuffledFullyConnectedWorkerImpl()
530 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input)); in ShuffledFullyConnectedWorkerImpl()
595 int8x16_t weights0 = vld1q_s8(shuffled_weights_ptr + 0); in ShuffledFullyConnectedWorkerImpl() local
607 local_accum0 = vmull_s8(vget_low_s8(weights0), vget_low_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
612 vmlal_s8(local_accum0, vget_high_s8(weights0), vget_high_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
|