| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op_gpu.cu.cc | 48 const int upper_diag_index, const T padding_value, in MatrixDiagKernel() argument
67 output_ptr[index] = padding_value; in MatrixDiagKernel()
79 const Eigen::Index max_diag_len, const T padding_value, in Compute()
96 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
105 const int upper_diag_index, const T padding_value, in MatrixDiagPartKernel() argument
128 output_ptr[index] = padding_value; in MatrixDiagPartKernel()
140 const Eigen::Index max_diag_len, const T padding_value, in Compute()
157 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
|
| D | matrix_diag_op.cc | 65 T padding_value(0); in Compute() local
89 padding_value = context->input(2).flat<T>()(0); in Compute()
140 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
171 T padding_value(0); in Compute() local
197 padding_value = context->input(4).flat<T>()(0); in Compute()
259 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
336 const Eigen::Index max_diag_len, const T padding_value, in Compute()
348 &padding_value, &left_align_superdiagonal, in Compute()
371 output(batch, i, j) = padding_value; in Compute()
392 const Eigen::Index max_diag_len, const T padding_value, in Compute()
[all …]
|
| D | matrix_diag_op.h | 54 const T padding_value, const bool left_align_superdiagonal,
65 const Eigen::Index max_diag_len, const T padding_value,
|
| /external/XNNPACK/src/subgraph/ |
| D | static-constant-pad.c | 23 float padding_value, in xnn_define_static_constant_pad() argument
56 node->params.static_pad.padding_value = fp32_to_bits(padding_value); in xnn_define_static_constant_pad()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 420 for padding_value, align in zip_to_first_list_length([0, 555, -11],
427 padding_value=padding_value,
430 solution = (solution + padding_value * mask).astype(dtype)
483 for padding_value, align in zip_to_first_list_length([0, 555, -11],
493 padding_value=padding_value,
496 solution = solution + padding_value * mask
512 padding_value=padding_value,
515 solution = solution + padding_value * mask
531 padding_value=padding_value,
534 solution = solution + padding_value * mask
[all …]
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 21 uint32_t padding_value, in create_constant_pad_nd() argument
46 constant_pad_op->pad_value = padding_value; in create_constant_pad_nd()
61 const void* padding_value, in xnn_create_constant_pad_nd_x32() argument
66 … *((uint32_t*) padding_value), flags, xnn_operator_type_constant_pad_nd_x32, constant_pad_op_out); in xnn_create_constant_pad_nd_x32()
149 .padding_value = constant_pad_op->pad_value, in setup_constant_pad_nd()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagPartV3.pbtxt | 17 name: "padding_value"
56 padding_value ; otherwise.
66 padding_value ; otherwise.
130 # padding_value = 9
131 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagPartV2.pbtxt | 17 name: "padding_value"
44 padding_value ; otherwise.
54 padding_value ; otherwise.
89 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagV2.pbtxt | 33 name: "padding_value"
68 padding_value ; otherwise
77 padding_value ; otherwise
129 # Rectangular matrix with inferred num_cols and padding_value = 9.
130 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| D | api_def_MatrixDiagV3.pbtxt | 33 name: "padding_value"
80 padding_value ; otherwise
89 padding_value ; otherwise
169 # Rectangular matrix with inferred num_cols and padding_value = 9.
170 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 221 padding_value=7,
240 padding_value=7,
257 padding_value=7,
266 padding_value=7,
276 padding_value=7,
287 padding_value=[0, 1],
296 padding_value=7,
305 padding_value=7,
314 padding_value=7,
324 padding_value=7,
[all …]
|
| D | matrix_diag_ops_test.py | 502 for padding_value, align in zip_to_first_list_length([555, -11],
507 solution = solution + (mask * padding_value)
514 "padding_value": padding_value,
714 for padding_value, align in zip_to_first_list_length([555, -11],
719 solution = solution + (mask * padding_value)
724 "padding_value": padding_value,
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 46 for (int64 padding_value : explicit_paddings) { in CheckValidPadding() local
47 if (padding_value < 0) { in CheckValidPadding()
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 146 const uint32_t padding_value = u32rng(); in TestX32() local
149 std::fill(output_ref.begin(), output_ref.end(), padding_value); in TestX32()
180 &padding_value, 0, &pad_op)); in TestX32()
210 << ", padding value = " << padding_value; in TestX32()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 260 xla::XlaOp padding_value = zero; in Compile() local
269 padding_value = context->Input(4); in Compile()
315 xla::XlaOp output = xla::Broadcast(padding_value, output_shape.dim_sizes()); in Compile()
371 xla::XlaOp padding_value = zero; in Compile() local
378 padding_value = context->Input(2); in Compile()
428 single_diag = xla::Pad(single_diag, padding_value, padding_config); in Compile()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | encoder_common.cc | 103 const int32_t padding_value, in CopyDataToTensorAndPadOrTruncate() argument
115 output_buffer[output_offset] = padding_value; in CopyDataToTensorAndPadOrTruncate()
|
| D | encoder_common.h | 62 const int32_t padding_value,
|
| /external/libvpx/libvpx/test/ |
| D | buffer.h | 93 void SetPadding(const T padding_value);
278 void Buffer<T>::SetPadding(const T padding_value) { in SetPadding() argument
280 padding_value_ = padding_value; in SetPadding()
284 src[i] = padding_value; in SetPadding()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorImagePatch.h | 72 … PaddingType padding_type, Scalar padding_value)
78 m_padding_type(padding_type), m_padding_value(padding_value) {}
86 Scalar padding_value)
93 m_padding_type(PADDING_VALID), m_padding_value(padding_value) {}
124 Scalar padding_value() const { return m_padding_value; }
179 m_paddingValue = op.padding_value();
|
| D | TensorVolumePatch.h | 67 … PaddingType padding_type, Scalar padding_value)
73 m_padding_type(padding_type), m_padding_value(padding_value) {}
82 Scalar padding_value)
89 m_padding_type(PADDING_VALID), m_padding_value(padding_value) {}
132 Scalar padding_value() const { return m_padding_value; }
192 m_paddingValue = op.padding_value();
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixDiagPartV2.pbtxt | 12 name: "padding_value"
|
| D | MatrixDiagV2.pbtxt | 20 name: "padding_value"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixDiagPartV2.pbtxt | 12 name: "padding_value"
|
| D | MatrixDiagV2.pbtxt | 20 name: "padding_value"
|
| D | MatrixDiagPartV3.pbtxt | 12 name: "padding_value"
|