Searched refs:input2_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 573 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
580 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
586 const T* input2_data, const Dims<4>& input2_dims, in Div() argument
593 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Div()
889 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
894 DimsToShape(input2_dims), input2_data, in Comparison()
902 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
918 DimsToShape(input2_dims), input2_data, in Comparison()
926 const Dims<4>& input2_dims, bool* output_data, in BroadcastComparison() argument
931 input1_data, DimsToShape(input2_dims), in BroadcastComparison()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 874 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
881 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
931 const float* input2_data, const Dims<4>& input2_dims, in Add() argument
940 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
948 const uint8* input2_data, const Dims<4>& input2_dims, in Add() argument
979 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
985 const int32* input2_data, const Dims<4>& input2_dims, in Add() argument
994 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
1000 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd() argument
1007 DimsToShape(input2_dims), input2_data, in BroadcastAdd()
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | unary_test.py | 43 input2_dims = [12, 5, 8, 1, 12, 1, 1]
78 dtype=dtype, shape=input2_dims, name=input2_name)
103 input_dims=[[input_dims, input2_dims]],
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3557 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
3564 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
|