| /external/tensorflow/tensorflow/core/kernels/ |
| 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/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/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/core/util/ |
| D | padding.cc | 53 for (int64 padding_value : explicit_paddings) { in CheckValidPadding() local
54 if (padding_value < 0) { in CheckValidPadding()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| 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/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 259 xla::XlaOp padding_value = zero; in Compile() local
268 padding_value = context->Input(4); in Compile()
314 xla::XlaOp output = xla::Broadcast(padding_value, output_shape.dim_sizes()); in Compile()
370 xla::XlaOp padding_value = zero; in Compile() local
377 padding_value = context->Input(2); in Compile()
427 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();
|
| D | TensorPadding.h | 63 …orPaddingOp(const XprType& expr, const PaddingDimensions& padding_dims, const Scalar padding_value)
64 : m_xpr(expr), m_padding_dims(padding_dims), m_padding_value(padding_value) {}
69 Scalar padding_value() const { return m_padding_value; }
104 : m_impl(op.expression(), device), m_padding(op.padding()), m_paddingValue(op.padding_value())
|
| D | TensorBase.h | 691 … const PaddingType padding_type = PADDING_SAME, const Scalar padding_value = Scalar(0)) const {
693 … in_row_stride, in_col_stride, 1, 1, padding_type, padding_value);
704 const Scalar padding_value) const { in extract_image_patches() argument
707 … padding_top, padding_bottom, padding_left, padding_right, padding_value); in extract_image_patches()
714 … const PaddingType padding_type = PADDING_SAME, const Scalar padding_value = Scalar(0)) const {
715 …, patch_cols, plane_stride, row_stride, col_stride, 1, 1, 1, 1, 1, 1, padding_type, padding_value);
726 …const Index padding_left, const Index padding_right, const Scalar padding_value = Scalar(0)) const…
727 …_top_z, padding_bottom_z, padding_top, padding_bottom, padding_left, padding_right, padding_value);
774 pad(const PaddingDimensions& padding, const Scalar padding_value) const { in pad() argument
775 … return TensorPaddingOp<const PaddingDimensions, const Derived>(derived(), padding, padding_value); in pad()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixDiagPartV2.pbtxt | 12 name: "padding_value"
|
| D | MatrixDiagV2.pbtxt | 20 name: "padding_value"
|
| D | MatrixDiagPartV3.pbtxt | 12 name: "padding_value"
|
| D | MatrixDiagV3.pbtxt | 20 name: "padding_value"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 386 ElementsAttr padding_value; in ConvertTFMatrixDiagV2orV3() local
387 if (!matchPattern(tf_matrix_diag_v2_or_v3_op.padding_value(), in ConvertTFMatrixDiagV2orV3()
388 m_Constant(&padding_value))) in ConvertTFMatrixDiagV2orV3()
390 for (auto value : padding_value.getValues<APInt>()) { in ConvertTFMatrixDiagV2orV3()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 59 HloInstruction* padding_value, in MakePadHlo() argument
62 CHECK_EQ(computation, padding_value->parent()); in MakePadHlo()
65 ShapeInference::InferPadShape(operand->shape(), padding_value->shape(), in MakePadHlo()
68 pad_shape, operand, padding_value, padding_config)); in MakePadHlo()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_tridiag.py | 354 padding_value=0.)
365 padding_value=0.)
|