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
26 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot()
27 HloInstruction *lhs = batch_dot->mutable_operand(0), in ElideDegenerateBatchDimensionFromBatchDot()
28 *rhs = batch_dot->mutable_operand(1); in ElideDegenerateBatchDimensionFromBatchDot()
74 batch_dot->precision_config())); in ElideDegenerateBatchDimensionFromBatchDot()
77 MakeReshapeHlo(batch_dot->shape(), new_dot)); in ElideDegenerateBatchDimensionFromBatchDot()
79 VLOG(2) << "Replaced " << batch_dot->ToString() << " with " in ElideDegenerateBatchDimensionFromBatchDot()
83 batch_dot->parent()->ReplaceInstruction(batch_dot, new_dot_reshaped)); in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | batch_dot_simplification.h | 35 HloInstruction* batch_dot);
|
| /external/tensorflow/tensorflow/contrib/keras/api/keras/backend/ |
| D | __init__.py | 29 from tensorflow.python.keras.backend import batch_dot
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 525 output = K.batch_dot(x1, x2, axes)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.backend.pbtxt | 36 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 | 1482 def batch_dot(x, y, axes=None): function
4830 output = batch_dot(x_aggregate, kernel)
|
| D | backend_test.py | 292 xy = keras.backend.batch_dot(x, y, axes=[1, 2])
|