Searched defs:BroadcastMul (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad.cc | 72 Status BroadcastMul(AbstractContext* ctx, AbstractTensorHandle* vec, in BroadcastMul() function
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 71 Output BroadcastMul(const Scope& scope, const Output& vec, const Output& mat) { in BroadcastMul() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1534 inline void BroadcastMul(const uint8* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
1552 inline void BroadcastMul(const uint8* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
1903 void BroadcastMul(const T* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
1917 void BroadcastMul(const T* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3731 inline void BroadcastMul(const uint8* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
3753 inline void BroadcastMul(const uint8* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
4748 void BroadcastMul(const T* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
4762 inline void BroadcastMul(const float* input1_data, const Dims<4>& input1_dims, in BroadcastMul() function
|