Searched refs:batch_dot (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batch_dot_simplification.cc | 25 HloInstruction* batch_dot) { in ElideDegenerateBatchDimensionFromBatchDot() argument
37 batch_dot->dot_dimension_numbers().lhs_batch_dimensions(), in ElideDegenerateBatchDimensionFromBatchDot()
38 batch_dot->dot_dimension_numbers().rhs_batch_dimensions()) || in ElideDegenerateBatchDimensionFromBatchDot()
40 batch_dot->dot_dimension_numbers().lhs_batch_dimensions()))) { in ElideDegenerateBatchDimensionFromBatchDot()
44 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot()
45 HloInstruction *lhs = batch_dot->mutable_operand(0), in ElideDegenerateBatchDimensionFromBatchDot()
46 *rhs = batch_dot->mutable_operand(1); in ElideDegenerateBatchDimensionFromBatchDot()
93 batch_dot->precision_config(), in ElideDegenerateBatchDimensionFromBatchDot()
94 /*preferred_element_type=*/batch_dot->shape().element_type())); in ElideDegenerateBatchDimensionFromBatchDot()
97 MakeReshapeHlo(batch_dot->shape(), new_dot)); in ElideDegenerateBatchDimensionFromBatchDot()
[all …]
|
| D | batch_dot_simplification.h | 35 HloInstruction* batch_dot);
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 703 output = K.batch_dot(x1, x2, axes)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1304 auto batch_dot = BatchDot(x, y); in XLA_TEST_P() local
1305 auto output_shape = builder.GetShape(batch_dot).ValueOrDie(); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.backend.pbtxt | 32 name: "batch_dot"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.backend.pbtxt | 36 name: "batch_dot"
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2020 def batch_dot(x, y, axes=None): function
5959 output = batch_dot(x_aggregate, kernel)
|
| D | backend_test.py | 386 xy = backend.batch_dot(x, y, axes=axes)
|